如何安装和配置NVIDIA Docker?
时间: 2024-11-03 20:15:53 浏览: 8
安装和配置NVIDIA Docker一般涉及以下几个步骤:
1. **安装NVIDIA Container Tools**:
- 对于Debian或Ubuntu系统,你可以从NVIDIA官方仓库安装,使用`apt-get`命令:
```
sudo apt-get update && sudo apt-get install nvidia-container-toolkit
```
- 对于RHEL/CentOS,可以添加NVIDIA软件源然后安装:
```bash
curl https://us.download.nvidia.com/tesla/daemon-tools/RHEL7/nvidia-container-runtime-deb-key | sudo apt-key add -
echo "deb http://us.download.nvidia.com/tesla/daemon-tools/RHEL7 ./" | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update && sudo apt-get install nvidia-container-toolkit
```
2. **安装Docker本身**(如果你还没有安装):
```
sudo apt-get install docker-ce docker-ce-cli containerd.io
```
3. **启动并启用NVIDIA Docker服务**:
```bash
sudo systemctl start nvidia-docker
sudo systemctl enable nvidia-docker
```
4. **验证安装**:
运行 `docker run --gpus all nvidia/cuda:latest nvidia-smi` 来检查GPU是否可以正常使用。
5. **配置Dockerfile**:
当创建包含GPU应用的Dockerfile时,需要指定使用`FROM nvidia`镜像,并在适当的地方添加`RUN`, `COPY`, 或`WORKDIR`指令来设置环境和安装依赖。
6. **构建和运行容器**:
使用`docker build`命令构建容器,然后使用`docker run`加上`--gpus=all`或具体的GPU ID来运行它。
记得每次登录新的终端,都需要运行`nvidia-docker exec`进入容器内部,因为默认只在当前会话生效。
阅读全文