diff --git a/site/en/install/docker.md b/site/en/install/docker.md index b46053a611..376ca0820a 100644 --- a/site/en/install/docker.md +++ b/site/en/install/docker.md @@ -130,7 +130,7 @@ lspci | grep -i nvidia Verify your `nvidia-docker` installation:
-docker run --gpus all --rm nvidia/cuda nvidia-smi
+docker run --rm --runtime=nvidia --gpus all ubuntu nvidia-smi
 
Note: `nvidia-docker` v2 uses `--runtime=nvidia` instead of `--gpus all`. `nvidia-docker` v1 uses the `nvidia-docker` alias,