centos 安装nvidia-docker
时间: 2024-05-06 09:13:24 浏览: 276
CentOS是一种流行的Linux操作系统,而NVIDIA Docker是一个用于在容器中运行GPU加速应用程序的工具。下面是在CentOS上安装NVIDIA Docker的步骤:
1. 首先,确保你的CentOS系统已经安装了Docker。你可以通过运行以下命令来检查:
```
docker --version
```
2. 如果你还没有安装Docker,可以按照以下步骤进行安装:
- 添加Docker的官方存储库:
```
sudo yum install -y yum-utils
sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
```
- 安装Docker引擎:
```
sudo yum install -y docker-ce docker-ce-cli containerd.io
```
- 启动Docker服务并设置开机自启:
```
sudo systemctl start docker
sudo systemctl enable docker
```
3. 安装NVIDIA驱动程序。你可以通过以下命令来安装适合你的NVIDIA显卡的驱动程序:
```
sudo yum install -y nvidia-driver-latest-dkms
```
4. 安装NVIDIA Container Toolkit。这是一个用于在Docker容器中运行GPU应用程序的工具包,其中包括nvidia-docker命令。
- 添加NVIDIA的官方存储库:
```
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | \
sudo tee /etc/yum.repos.d/nvidia-docker.repo
```
- 安装NVIDIA Container Toolkit:
```
sudo yum install - 重新加载Docker配置并重启Docker服务:
```
sudo systemctl restart docker
```
5. 验证安装是否成功。你可以运行以下命令来验证nvidia-docker是否正确安装:
```
sudo docker run --gpus all nvidia/cuda:11.0-base nvidia-smi
```
如果一切正常,你应该能够看到NVIDIA显卡的信息。
阅读全文