Configurar o Visual Studio Code local - SageMaker Inteligência Artificial da Amazon
Configurar 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 a SageMaker espaços com acesso remoto, você poderá conectar seu Visual Studio Code local aos seus SageMaker espaços remotos.

Tópicos

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 a SageMaker espaços com acesso remoto.

Conecte-se ao seu VS Code local

Antes de 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ê.

Use o procedimento a seguir para estabelecer uma conexão usando o link direto.

  1. Inicie o Amazon SageMaker Studio.

  2. Na interface do usuário do Studio, navegue até seu espaço.

  3. Escolha Abrir espaço com.

  4. Escolha VS Code. Ao fazer isso, você pode ser solicitado a abrir o Visual Studio Code. Quando você escolhe 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 o procedimento a seguir para estabelecer uma conexão usando AWS Toolkit for Visual Studio Code o.

  1. Abra o VS Code.

  2. Abra a extensão AWS Toolkit.

  3. Conecte-se AWS a.

  4. No AWS Explorer, expanda a SageMaker IA.

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

  6. Escolha o ícone Connect ao lado do seu espaço para iniciá-lo.

    nota

    • Pare e reinicie o espaço no Toolkit for Visual Studio para habilitar o acesso remoto, se ainda não estiver conectado.

    • Se o espaço não estiver usando um tamanho de instância compatível, você deverá alterar a instância.

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

Use o procedimento a seguir para estabelecer uma conexão usando a CLI SSH.

Para essa configuração, verifique se você tem o AWS Command Line Interface (AWS CLI) e o plug-in do Gerenciador de Sessões para o AWS CLI instalado.

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, eles PARTITION podem 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/