Configurar o Visual Studio Code local - SageMaker IA da Amazon
Habilite o acesso remoto ao seu espaçoConfigurar o ambiente localConecte-se ao seu VS Code local

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

Configurar o Visual Studio Code local

Depois que os administradores concluírem as instruçõesConecte seu Visual Studio Code local aos SageMaker espaços, você poderá conectar seu Visual Studio Code local aos seus SageMaker espaços remotos.

Habilite o acesso remoto ao seu espaço

nota

Essa configuração só precisa ser feita uma vez.

Para um espaço existente

Ative o acesso remoto ao seu SageMaker espaço:

  1. Abra a SageMaker AI UI.

  2. Navegue até o seu espaço.

  3. Nos detalhes do espaço, ative o Acesso remoto.

  4. Comece seu espaço.

Configurar o ambiente local

Instale o Visual Studio Code em sua máquina local. Para obter informações sobre os requisitos, consulteConecte seu Visual Studio Code local aos SageMaker espaços.

Conecte-se ao seu VS Code local

Antes que você possa estabelecer uma conexão do Visual Studio Code local com seus SageMaker espaços remotos, seu administrador deveConfigurar o acesso remoto. Seu administrador configura um método específico para você estabelecer uma conexão. Escolha o método que foi configurado para você.

  1. Na SageMaker AI UI, navegue até seu espaço

  2. Escolha Espaço aberto com

  3. Escolha VS Code. Ao fazer isso, você pode ser solicitado a abrir o Visual Studio Code. Quando você opta por fazer isso, seu VS Code local é aberto com outro pop-up para confirmar. Depois de concluída, a conexão remota é estabelecida.

Método 2: AWS Toolkit for Visual Studio Code

Use as informações e instruções contidas AWS no guia do usuário do Toolkit for VS Code.

  1. Abra o VS Code.

  2. Abra a extensão AWS Toolkit.

  3. No AWS Explorer, expanda a SageMaker IA.

  4. Encontre o espaço do seu estúdio.

  5. Escolha o ícone Connect ao lado do seu espaço.

Método 3: Conecte-se a partir do terminal via SSH CLI

Crie um script de shell (por exemplo,/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. Torne o script executável:

    chmod +x /home/user/sagemaker_connect.sh

  2. Configure $HOME/.ssh/config para adicionar a seguinte 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 exemplo, PARTITION pode seraws.

Se você precisar usar um perfil de AWS credencial nomeado, altere o comando proxy da seguinte forma:

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

  • Conecte-se via SSH ou execute o comando SCP:

ssh my_space_name_abc
scp file_abc my_space_name_abc:/tmp/
Tópicos