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 los espacios, podrá conectar el código de Visual Studio local a SageMaker los espacios remotos.

Habilite el acceso remoto a su espacio

Para un espacio existente

Activa el acceso remoto a tu SageMaker espacio:

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 los espacios.

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.

Método 1: enlace profundo desde la interfaz de usuario de Studio

Método 2: AWS Toolkit for Visual Studio Code

Utilice la información y las instrucciones de connect to AWS en la guía del usuario del Toolkit for VS Code.

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

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"

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

ssh my_space_name_abc
scp file_abc my_space_name_abc:/tmp/