Les traductions sont fournies par des outils de traduction automatique. En cas de conflit entre le contenu d'une traduction et celui de la version originale en anglais, la version anglaise prévaudra.
Configurer Remote IDE
Une fois que les administrateurs ont terminé les instructionsConnectez votre IDE distant à des SageMaker espaces avec accès à distance, vous pouvez connecter votre IDE distant à vos SageMaker espaces distants.
Configuration de votre environnement local
Installez votre IDE distant préféré sur votre machine locale :
Pour plus d'informations sur les exigences de version, consultezExigences relatives à la version IDE.
Connectez-vous à votre IDE distant
Avant de pouvoir établir une connexion entre votre IDE distant et vos SageMaker espaces distants, votre administrateur doitConfiguration de l’accès à distance. Votre administrateur configure une méthode spécifique pour vous permettre d’établir une connexion. Choisissez la méthode qui a été configurée pour vous.
Méthode 1 : lien profond depuis l’interface utilisateur de Studio
Utilisez la procédure suivante pour établir une connexion à l’aide d’un lien profond.
-
Lancez Amazon SageMaker Studio.
-
Dans l’interface utilisateur de Studio, accédez à votre espace.
-
Choisissez Ouvrir dans VS Code, Ouvrir dans Kiro ou Ouvrir dans le curseur pour votre IDE préféré. Assurez-vous que votre IDE préféré est déjà installé sur votre ordinateur local.
-
Lorsque vous y êtes invité, confirmez l'ouverture de votre IDE. Votre IDE s'ouvre avec une autre fenêtre contextuelle pour confirmer. Une fois terminée, la connexion à distance est établie.
Utilisez la procédure suivante pour établir une connexion à l’aide d’ AWS Toolkit for Visual Studio Code. Cette méthode est disponible pour VS Code, Kiro et Cursor.
-
Ouvrez votre IDE distant (VS Code, Kiro ou Cursor).
-
Ouvrez l'extension AWS Toolkit.
-
Connectez-vous à AWS.
-
Dans l' AWS explorateur, développez SageMaker l'IA, puis développez Studio.
-
Recherchez votre espace Studio.
-
Choisissez l’icône Connexion à côté de votre espace pour le démarrer.
-
Arrêtez et redémarrez l’espace dans Toolkit for Visual Studio pour permettre l’accès à distance, si vous n’êtes pas déjà connecté.
-
Si l’espace n’utilise pas une taille d’instance prise en charge, il vous sera demandé de changer l’instance.
Méthode 3 : Connexion depuis le terminal via la CLI SSH
Choisissez l'une des options de plate-forme suivantes pour afficher la procédure d'établissement d'une connexion à l'aide de la CLI SSH.
- Linux/macOS
-
Créez un script shell (par exemple, /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"
-
Rendez le script exécutable :
chmod +x /home/user/sagemaker_connect.sh
-
Configurez $HOME/.ssh/config pour ajouter l’entrée suivante :
Host space-name
HostName 'arn:PARTITION:sagemaker:us-east-1:111122223333:space/domain-id/space-name'
ProxyCommand '/home/user/sagemaker_connect.sh' '%h'
ForwardAgent yes
AddKeysToAgent yes
StrictHostKeyChecking accept-new
Par exemple, le paramètre PARTITION peut avoir la valeur aws.
Si vous devez utiliser un profil AWS d'identification nommé, modifiez la commande proxy comme suit :
ProxyCommand '/home/user/sagemaker_connect.sh' '%h' YOUR_CREDENTIAL_PROFILE_NAME
ssh space-name
scp file_abc space-name:/tmp/
- Windows
-
Prérequis pour Windows :
Créez un PowerShell script (par exemple,C:\Users\user-name\sagemaker_connect.ps1) :
# sagemaker_connect.ps1
param(
[Parameter(Mandatory=$true)]
[string]$SpaceArn,
[Parameter(Mandatory=$false)]
[string]$AwsProfile = ""
)
# Enable error handling
$ErrorActionPreference = "Stop"
# Validate ARN and extract region
if ($SpaceArn -match "^arn:aws[-a-z]*:sagemaker:([a-z0-9-]+):[0-9]{12}:space\/[^\/]+\/[^\/]+$") {
$AwsRegion = $Matches[1]
} else {
Write-Error "Error: Invalid SageMaker Studio Space ARN format."
exit 1
}
# Build AWS CLI command
$awsCommand = @("sagemaker", "start-session", "--resource-identifier", $SpaceArn, "--region", $AwsRegion)
if ($AwsProfile) {
$awsCommand += @("--profile", $AwsProfile)
}
try {
# Start session and capture output
Write-Host "Starting SageMaker session..." -ForegroundColor Green
$startSessionOutput = & aws @awsCommand
# Try to parse JSON response
try {
$sessionData = $startSessionOutput | ConvertFrom-Json
} catch {
Write-Error "Failed to parse JSON response: $_"
Write-Host "Raw response was:" -ForegroundColor Yellow
Write-Host $startSessionOutput
exit 1
}
$sessionId = $sessionData.SessionId
$streamUrl = $sessionData.StreamUrl
$token = $sessionData.TokenValue
# Validate extracted values
if (-not $sessionId -or -not $streamUrl -or -not $token) {
Write-Error "Error: Failed to extract session information from sagemaker start session response."
Write-Host "Parsed response was:" -ForegroundColor Yellow
Write-Host ($sessionData | ConvertTo-Json)
exit 1
}
Write-Host "Session started successfully. Connecting..." -ForegroundColor Green
# Create session manager plugin command
$sessionJson = @{
streamUrl = $streamUrl
tokenValue = $token
sessionId = $sessionId
} | ConvertTo-Json -Compress
# Escape the JSON string
$escapedJson = $sessionJson -replace '"', '\"'
# Call session-manager-plugin
& session-manager-plugin "$escapedJson" $AwsRegion "StartSession"
} catch {
Write-Error "Failed to start session: $_"
exit 1
}
Host space-name
HostName "arn:aws:sagemaker:us-east-1:111122223333:space/domain-id/space-name"
ProxyCommand "C:\WINDOWS\System32\WindowsPowerShell\v1.0\powershell.exe" -ExecutionPolicy RemoteSigned -File "C:\\Users\\user-name\\sagemaker_connect.ps1" "%h"
ForwardAgent yes
AddKeysToAgent yes
User sagemaker-user
StrictHostKeyChecking accept-new