Ubuntu24.04安装nvidia-docker
时间: 2024-09-25 16:19:42 浏览: 614
Ubuntu 24.04 ( focal fossa ) 使用的是新版的软件包管理系统,其官方并不直接支持 NVIDIA-Docker 的预装。NVIDIA-Docker是一个针对深度学习和GPU计算优化的Docker容器插件,它允许你在基于NVIDIA GPU的容器中运行CUDA应用。
以下是安装 NVIDIA-Docker 在 Ubuntu 24.04 上的步骤:
1. **安装依赖**:
首先需要安装`apt-transport-https`和`curl`,因为NVIDIA Docker的安装依赖于HTTPS仓库:
```
sudo apt update
sudo apt install apt-transport-https curl software-properties-common
```
2. **添加NVIDIA GPG密钥**:
```
curl -fsSL https://download.nvidia.com/debian/gpg/nvidia-gpg | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-archive-keyring.gpg
```
3. **添加NVIDIA的APT源**:
```
echo "deb [arch=amd64 signed-by=/usr/share/keyrings/nvidia-archive-keyring.gpg] https://us.download.nvidia.com/repos/ubuntu/focal nvidia-x86_64 contrib" | sudo tee /etc/apt/sources.list.d/nvidia.list > /dev/null
```
4. **更新并安装NVIDIA-Docker**:
```
sudo apt-get update
sudo apt-get install nvidia-container-toolkit
```
5. **重启Docker服务**:
```
sudo systemctl restart docker
```
6. **验证安装**:
最后,你可以通过检查 `docker run --gpus all nvidia/cuda:latest nvidia-smi` 来确认NVIDIA-Docker是否成功安装并能识别到GPU。
阅读全文