NPU Inference/Training Test on SSH Server

Arguments

• $0 (ssh-host): SSH connection string, e.g., user@192.168.1.100 or just hostname (required)
• $1 (image): Docker image name. If not provided, auto-detect the highest version vllm-ascend image on the remote machine
• $2 (devices): NPU devices to use, e.g., 0,1,2,3. If not provided, detect available devices
• $3 (test-command): The command to run inside the container. If not provided, enter interactive mode

Instructions

Step 1: Test SSH Connection