nvidia-docker 安装教程
时间: 2023-09-13 19:06:42 浏览: 151
要安装 nvidia-docker,您可以按照以下步骤进行操作:
1. 确保您的计算机上已经正确安装了 NVIDIA 驱动程序。您可以使用以下命令来检查:
```shell
nvidia-smi
```
如果能够正确显示 NVIDIA GPU 信息,则表示驱动程序已经成功安装。
2. 安装 Docker CE。您可以按照 Docker 官方文档的说明进行安装。取决于您的操作系统,可以参考以下链接:
- [Ubuntu 安装 Docker CE](https://docs.docker.com/engine/install/ubuntu/)
- [CentOS 安装 Docker CE](https://docs.docker.com/engine/install/centos/)
- [Windows 安装 Docker Desktop](https://docs.docker.com/desktop/windows/install/)
- [Mac 安装 Docker Desktop](https://docs.docker.com/desktop/mac/install/)
3. 添加 NVIDIA 的 apt 仓库密钥。使用以下命令将密钥添加到系统中:
```shell
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
```
4. 添加 NVIDIA 的 apt 仓库。根据您的操作系统版本选择合适的命令,例如:
- Ubuntu 20.04:
```shell
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
```
- Ubuntu 18.04 / 16.04 或 Debian:
```shell
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
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
```
5. 安装 nvidia-docker2 包。使用以下命令进行安装:
```shell
sudo apt-get update
sudo apt-get install -y nvidia-docker2
```
6. 重新启动 Docker 服务:
```shell
sudo systemctl restart docker
```
7. 确认 nvidia-docker 是否成功安装。您可以运行以下命令来验证:
```shell
docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
```
如果能够正确显示 NVIDIA GPU 信息,则表示 nvidia-docker 安装成功。
请注意,上述步骤是针对 Ubuntu 和 Debian 系统的说明。如果您使用的是其他操作系统,请参考 NVIDIA 官方文档的指南进行安装。
希望这对您有所帮助!如有任何问题,请随时提问。