Configurer le code Visual Studio local - Amazon SageMaker AI
Configurez votre environnement localConnectez-vous à votre VS Code local

Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.

Configurer le code Visual Studio local

Une fois que les administrateurs ont terminé les instructionsConnectez votre code Visual Studio local à des SageMaker espaces avec accès à distance, vous pouvez connecter votre code Visual Studio local à vos SageMaker espaces distants.

Rubriques

Configurez votre environnement local

Installez Visual Studio Code sur votre machine locale. Pour plus d'informations sur les exigences, voirConnectez votre code Visual Studio local à des SageMaker espaces avec accès à distance.

Connectez-vous à votre VS Code local

Avant de pouvoir établir une connexion entre votre code Visual Studio local et vos SageMaker espaces distants, votre administrateur doit le faireConfigurer l'accès à distance. Votre administrateur définit une méthode spécifique pour établir une connexion. Choisissez la méthode qui a été configurée pour vous.

Pour établir une connexion à l'aide d'un lien profond, procédez comme suit.

  1. Lancez Amazon SageMaker Studio.

  2. Dans l'interface utilisateur de Studio, accédez à votre espace.

  3. Choisissez Ouvrir un espace avec.

  4. Choisissez VS Code. Lorsque vous le faites, vous pouvez être invité à ouvrir Visual Studio Code. Lorsque vous choisissez de le faire, votre code VS local s'ouvre avec une autre fenêtre contextuelle pour confirmer. Une fois terminée, la connexion à distance est établie.

Méthode 2 : AWS Toolkit for Visual Studio Code

Utilisez la procédure suivante pour établir une connexion à l'aide du AWS Toolkit for Visual Studio Code.

  1. Ouvrez VS Code.

  2. Ouvrez l'extension AWS Toolkit.

  3. Connectez-vous à AWS.

  4. Dans l' AWS explorateur, développez SageMaker l'IA.

  5. Trouvez votre espace de studio.

  6. Cliquez sur l'icône Connect à côté de votre espace pour le démarrer.

    Note

    • Arrêtez et redémarrez l'espace dans le Toolkit for Visual Studio pour permettre l'accès à distance, si vous n'êtes pas déjà connecté.

    • Si l'espace n'utilise pas une taille d'instance prise en charge, il vous sera demandé de modifier l'instance.

Méthode 3 : Connexion depuis le terminal via la CLI SSH

Utilisez la procédure suivante pour établir une connexion à l'aide de la CLI SSH.

Pour cette configuration, assurez-vous que le plugin AWS Command Line Interface (AWS CLI) et le plug-in Session Manager AWS CLI sont installés.

Créez un script shell (par exemple,/home/user/sagemaker_connect.sh) :

#!/bin/bash
# Disable the -x option if printing each command is not needed.
set -exuo pipefail

SPACE_ARN="$1"
AWS_PROFILE="${2:-}"

# Validate ARN and extract region
if [[ "$SPACE_ARN" =~ ^arn:aws[-a-z]*:sagemaker:([a-z0-9-]+):[0-9]{12}:space\/[^\/]+\/[^\/]+$ ]]; then
    AWS_REGION="${BASH_REMATCH[1]}"
else
    echo "Error: Invalid SageMaker Studio Space ARN format."
    exit 1
fi

# Optional profile flag
PROFILE_ARG=()
if [[ -n "$AWS_PROFILE" ]]; then
    PROFILE_ARG=(--profile "$AWS_PROFILE")
fi

# Start session
START_SESSION_JSON=$(aws sagemaker start-session \
    --resource-identifier "$SPACE_ARN" \
    --region "${AWS_REGION}" \
    "${PROFILE_ARG[@]}")

# Extract fields using grep and sed
SESSION_ID=$(echo "$START_SESSION_JSON" | grep -o '"SessionId": "[^"]*"' | sed 's/.*: "//;s/"$//')
STREAM_URL=$(echo "$START_SESSION_JSON" | grep -o '"StreamUrl": "[^"]*"' | sed 's/.*: "//;s/"$//')
TOKEN=$(echo "$START_SESSION_JSON" | grep -o '"TokenValue": "[^"]*"' | sed 's/.*: "//;s/"$//')

# Validate extracted values
if [[ -z "$SESSION_ID" || -z "$STREAM_URL" || -z "$TOKEN" ]]; then
    echo "Error: Failed to extract session information from sagemaker start session response."
    exit 1
fi

# Call session-manager-plugin
session-manager-plugin \
    "{\"streamUrl\":\"$STREAM_URL\",\"tokenValue\":\"$TOKEN\",\"sessionId\":\"$SESSION_ID\"}" \
    "$AWS_REGION" "StartSession"

  1. Rendez le script exécutable :

    chmod +x /home/user/sagemaker_connect.sh

  2. Configurez $HOME/.ssh/config pour ajouter l'entrée suivante :

Host my_space_name_abc
  HostName 'arn:PARTITION:sagemaker:REGION:ACCOUNT:space/DOMAIN_ID/space_name_abc'
  ProxyCommand '/home/user/sagemaker_connect.sh' '%h'
  ForwardAgent yes
  AddKeysToAgent yes
  StrictHostKeyChecking accept-new

Par exemple, ils PARTITION peuvent êtreaws.

Si vous devez utiliser un profil AWS d'identification nommé, modifiez la commande proxy comme suit :

  ProxyCommand '/home/user/sagemaker_connect.sh' '%h' YOUR_CREDENTIAL_PROFILE_NAME

  • Connectez-vous via SSH ou exécutez la commande SCP :

ssh my_space_name_abc
scp file_abc my_space_name_abc:/tmp/