Configura il codice locale di Visual Studio - Amazon SageMaker AI
Abilita l'accesso remoto per il tuo spazioConfigura il tuo ambiente localeConnect al tuo VS Code locale

Le traduzioni sono generate tramite traduzione automatica. In caso di conflitto tra il contenuto di una traduzione e la versione originale in Inglese, quest'ultima prevarrà.

Configura il codice locale di Visual Studio

Dopo che gli amministratori hanno completato le istruzioni contenute inConnect il codice di Visual Studio locale agli SageMaker spazi, puoi connettere il codice di Visual Studio locale agli SageMaker spazi remoti.

Abilita l'accesso remoto per il tuo spazio

Nota

Questa configurazione deve essere eseguita una sola volta.

Per uno spazio esistente

Attiva l'accesso remoto per il tuo SageMaker spazio:

  1. Apri l'interfaccia utente SageMaker AI.

  2. Accedi al tuo spazio.

  3. Nei dettagli dello spazio, attiva Accesso remoto.

  4. Avvia il tuo spazio.

Configura il tuo ambiente locale

Installa Visual Studio Code sul tuo computer locale. Per informazioni sui requisiti, consultaConnect il codice di Visual Studio locale agli SageMaker spazi.

Connect al tuo VS Code locale

Prima di poter stabilire una connessione dal codice di Visual Studio locale agli SageMaker spazi remoti, l'amministratore deve farloConfigura l'accesso remoto. L'amministratore imposta un metodo specifico per stabilire una connessione. Scegli il metodo che è stato configurato per te.

  1. Nell'interfaccia utente SageMaker AI, accedi al tuo spazio

  2. Scegli Spazio aperto con

  3. Scegli VS Code. Quando lo fai, ti potrebbe essere richiesto di aprire Visual Studio Code. Quando scegli di farlo, il tuo VS Code locale si apre con un altro pop-up di conferma. Una volta completata, viene stabilita la connessione remota.

Metodo 2: AWS Toolkit for Visual Studio Code

Usa le informazioni e le istruzioni in Connect to AWS nella guida per l'utente di Toolkit for VS Code.

  1. Apri VS Code.

  2. Apri l'estensione AWS Toolkit.

  3. In AWS Explorer, espandi SageMaker AI.

  4. Trova il tuo spazio Studio.

  5. Scegli l'icona Connect accanto al tuo spazio.

Metodo 3: Connect dal terminale tramite SSH CLI

Crea uno script di shell (ad esempio,/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. Rendi eseguibile lo script:

    chmod +x /home/user/sagemaker_connect.sh

  2. Configura $HOME/.ssh/config per aggiungere la seguente voce:

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

Ad esempio, PARTITION può essereaws.

Se è necessario utilizzare un profilo di AWS credenziali denominato, modificare il comando proxy come segue:

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

  • Connect via SSH o esegui il comando SCP:

ssh my_space_name_abc
scp file_abc my_space_name_abc:/tmp/
Argomenti