在Ubuntu系统中，为使Docker能利用Nvidia GPU，需要安装Nvidia-docker2。首先，通过添加Nvidia-docker的GPG密钥和软件源，然后更新并安装Nvidia-docker2。安装完成后，重启docker服务。最后，通过运行带有--runtime=nvidia选项的nvidia/cuda镜像来验证GPU支持是否正常工作。
ubuntu安装Nvidia-docker2
在docker中要使用GPU，需要安装Nvidia-docker2才可以，以下是详细过程：
添加源
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
sudo apt-key add -
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
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd
测试
sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
扫一扫
5465
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
