Configure el código local de Visual Studio - Amazon SageMaker AI
Configure su entorno localConéctese a su VS Code local

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Configure el código local de Visual Studio

Una vez que los administradores hayan completado las instruccionesConecta tu código local de Visual Studio a SageMaker espacios con acceso remoto, podrá conectar el código de Visual Studio local a SageMaker los espacios remotos.

Temas

Configure su entorno local

Instale Visual Studio Code en su máquina local. Para obtener información sobre los requisitos, consulteConecta tu código local de Visual Studio a SageMaker espacios con acceso remoto.

Conéctese a su VS Code local

Antes de poder establecer una conexión desde su Visual Studio Code local a sus SageMaker espacios remotos, su administrador debe hacerloConfigura el acceso remoto. El administrador configura un método específico para que usted establezca una conexión. Elija el método que se configuró para usted.

Utilice el siguiente procedimiento para establecer una conexión mediante un enlace profundo.

  1. Abre Amazon SageMaker Studio.

  2. En la interfaz de usuario de Studio, navega hasta tu espacio.

  3. Elige Abrir espacio con.

  4. Elige VS Code. Cuando lo haga, es posible que se le pida que abra Visual Studio Code. Si decide hacerlo, su VS Code local se abre con otra ventana emergente para confirmarlo. Una vez completada, se establece la conexión remota.

Método 2: AWS Toolkit for Visual Studio Code

Utilice el siguiente procedimiento para establecer una conexión mediante el AWS Toolkit for Visual Studio Code.

  1. Abra VS Code.

  2. Abra la extensión AWS Toolkit.

  3. Conectarse a AWS.

  4. En el AWS Explorador, expande la SageMaker IA.

  5. Encuentra tu espacio de estudio.

  6. Selecciona el icono Connect situado junto a tu espacio para iniciarlo.

    nota

    • Detenga y reinicie el espacio del Toolkit for Visual Studio para habilitar el acceso remoto, si aún no está conectado.

    • Si el espacio no utiliza un tamaño de instancia compatible, se le pedirá que cambie la instancia.

Método 3: Conectarse desde el terminal a través de SSH CLI

Utilice el siguiente procedimiento para establecer una conexión mediante la CLI de SSH.

Para esta configuración, asegúrese de tener AWS CLI instalados el AWS Command Line Interface (AWS CLI) y el complemento Session Manager.

Cree un script de shell (por ejemplo,/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. Haga que el script sea ejecutable:

    chmod +x /home/user/sagemaker_connect.sh

  2. $HOME/.ssh/configConfigúrelo para añadir la siguiente entrada:

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

Por ejemplo, PARTITION puede seraws.

Si necesita usar un perfil de AWS credenciales con nombre, cambie el comando proxy de la siguiente manera:

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

  • Conéctese mediante SSH o ejecute el comando SCP:

ssh my_space_name_abc
scp file_abc my_space_name_abc:/tmp/