Configura il codice locale di Visual Studio - Amazon SageMaker AI
Configura 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 con accesso remoto, puoi connettere il codice di Visual Studio locale agli SageMaker spazi remoti.

Argomenti

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 con accesso remoto.

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.

Utilizza la seguente procedura per stabilire una connessione tramite deep link.

  1. Avvia Amazon SageMaker Studio.

  2. Nell'interfaccia utente di Studio, accedi al tuo spazio.

  3. Scegli Spazio aperto con.

  4. 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

Utilizzare la procedura seguente per stabilire una connessione utilizzando AWS Toolkit for Visual Studio Code.

  1. Apri VS Code.

  2. Apri l'estensione AWS Toolkit.

  3. Connect a AWS.

  4. In AWS Explorer, espandi l'SageMaker IA.

  5. Trova il tuo spazio Studio.

  6. Scegli l'icona Connect accanto al tuo spazio per avviarlo.

    Nota

    • Arresta e riavvia lo spazio nel Toolkit for Visual Studio per abilitare l'accesso remoto, se non è già connesso.

    • Se lo spazio non utilizza una dimensione di istanza supportata, ti verrà chiesto di modificare l'istanza.

Metodo 3: Connect dal terminale tramite SSH CLI

Utilizzare la procedura seguente per stabilire una connessione utilizzando la CLI SSH.

Per questa configurazione, assicuratevi di avere installato il plugin AWS Command Line Interface (AWS CLI) e il relativo plugin Session Manager. AWS 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/