Mengatur Kode Studio Visual - Amazon SageMaker AI
Siapkan lingkungan AndaConnect ke VS Code lokal Anda

Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Mengatur Kode Studio Visual

Setelah administrator menyelesaikan instruksiHubungkan Kode Visual Studio lokal Anda ke SageMaker spasi dengan akses jarak jauh, Anda dapat menghubungkan Kode Visual Studio lokal Anda ke SageMaker ruang jarak jauh Anda.

Topik

Siapkan lingkungan Anda

Instal Visual Studio Code di mesin lokal Anda. Untuk informasi tentang jaringan yang diperlukan, lihat Hubungkan Kode Visual Studio lokal Anda ke SageMaker spasi dengan akses jarak jauh.

Connect ke VS Code lokal Anda

Sebelum Anda dapat membuat koneksi dari Kode Visual Studio lokal Anda ke SageMaker ruang jarak jauh Anda, administrator Anda harusMengatur akses jarak jauh. Administrator Anda menyiapkan metode khusus bagi Anda untuk membuat koneksi. Pilih metode yang disiapkan untuk Anda.

Gunakan prosedur berikut untuk membuat koneksi menggunakan deep link.

  1. Luncurkan Amazon SageMaker Studio.

  2. Di UI Studio, navigasikan ke ruang Anda.

  3. Pilih Ruang terbuka dengan.

  4. Pilih Kode VS. Ketika Anda melakukannya, Anda mungkin diminta untuk Membuka Kode Visual Studio. Ketika Anda memilih untuk melakukannya, Kode VS lokal Anda terbuka dengan pop-up lain untuk mengonfirmasi. Setelah selesai, koneksi jarak jauh dibuat.

Metode 2:

Gunakan prosedur berikut untuk membuat koneksi menggunakan AWS Toolkit for Visual Studio Code.

  1. Kode VS Buka.

  2. Buka ekstensi AWS Toolkit.

  3. Sambungkan ke

  4. Di AWS Explorer, perluas SageMaker AI.

  5. Temukan ruang Studio Anda.

  6. Pilih ikon Connect di sebelah ruang Anda untuk memulainya.

    catatan

    • Hentikan dan restart ruang di Toolkit for Visual Studio untuk mengaktifkan akses jarak jauh, jika belum terhubung.

    • Jika ruang tidak menggunakan ukuran instans yang didukung, Anda akan diminta untuk mengubah instance.

Metode 3: Connect dari terminal melalui SSH CLI

Gunakan prosedur berikut untuk membuat koneksi menggunakan SSH CLI.

Untuk pengaturan ini, pastikan bahwa Anda memiliki AWS Command Line Interface (AWS CLI) dan plugin Session Manager untuk AWS CLI diinstal.

Buat skrip shell (misalnya,/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. Jadikan skrip dapat dieksekusi:

    chmod +x /home/user/sagemaker_connect.sh

  2. Konfigurasikan $HOME/.ssh/config untuk menambahkan entri berikut:

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

Misalnya, PARTITION bisaaws.

Jika Anda perlu menggunakan profil AWS kredensi bernama, ubah perintah proxy sebagai berikut:

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

  • Connect melalui SSH atau jalankan perintah SCP:

ssh my_space_name_abc
scp file_abc my_space_name_abc:/tmp/