要在Ubuntu上安装NVIDIA Docker，请按照以下步骤进行操作：
- 首先需要安装Docker CE和NVIDIA驱动程序。您可以使用以下命令来安装它们：
sudo apt-get update
sudo apt-get install docker-ce
sudo apt-get install nvidia-driver-<version>
其中，<version>是您要安装的NVIDIA驱动程序版本号。
- 安装NVIDIA Container Toolkit。您可以使用以下命令来安装它：
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
请注意，此命令将配置您的APT源以获取NVIDIA Docker软件包，并在安装后重启Docker。
- 确认NVIDIA Docker是否正确安装。您可以使用以下命令来运行一个简单的测试容器：
sudo docker run --gpus all nvidia/cuda:10.0-base nvidia-smi
如果一切正常，则应该看到类似于以下内容的输出：
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 418.87.01 Driver Version: 418.87.01 CUDA Version: 10.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 970M Off | 00000000:01:00.0 Off | N/A |
| N/A 46C P8 6W / N/A | 110MiB / 3015MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1037 G /usr/lib/xorg/Xorg 39MiB |
| 0 1429 G compiz 68MiB |
+-----------------------------------------------------------------------------+
这将显示您的NVIDIA GPU，并验证NVIDIA Docker是否正确配置。
恭喜您，现在您已成功地在Ubuntu上安装了NVIDIA Docker！
