ubuntu20.04docker cuda
时间: 2025-01-13 12:46:06 浏览: 49
### 配置 Docker 支持 CUDA
#### 设置 Docker 开机自启
为了确保每次启动系统时自动启用 Docker 服务,执行命令:
```bash
sudo systemctl enable docker
```
这一步骤能保证系统的持续性和稳定性[^1]。
#### 安装 NVIDIA Container Toolkit
为了让 Docker 能够利用 GPU 加速功能,需先安装 NVIDIA Container Toolkit。具体操作如下:
- 获取并添加 NVIDIA 的软件包仓库密钥:
```bash
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
```
- 将对应的发行版信息加入到 `/etc/apt/sources.list.d/` 文件夹下创建的新列表文件 `nvidia-docker.list` 中:
```bash
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
```
- 更新本地 APT 缓存以反映新添加的内容,并安装必要的组件:
```bash
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
```
完成上述步骤之后重启 Docker 服务使更改生效:
```bash
sudo systemctl restart docker
```
此时已经成功设置了 Docker 对 CUDA 的支持环境[^2]。
#### 测试配置是否正常工作
可以通过拉取一个带有 CUDA 工具链的基础镜像来验证整个流程是否正确无误地完成了设置。例如,尝试运行下面这条命令查看是否有可用的 GPU 设备被识别:
```bash
docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
```
如果一切顺利的话,则会看到类似于主机端输出的信息显示当前正在使用的显卡详情以及驱动版本等内容。
阅读全文
相关推荐


















