Docker笔记--Ubuntu20.04安装Nvidia-Docker

目录

1--添加软件包存储库

2--安装Nvidia-Docker2

3--测试

1--添加软件包存储库

1.  
   curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
2.  
    
3.  
   distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
4.  
    
5.  
   curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
6.  
    
7.  
   sudo apt-get update

2--安装Nvidia-Docker2

        安装 nvidia-docker2 软件包并重新加载 docker 守护程序配置：

1.  
   sudo apt-get install -y nvidia-docker2
2.  
    
3.  
   sudo pkill -SIGHUP dockerd

        执行 sudo apt-get install -y nvidia-docker2，可能会出现以下错误：E: Unable to locate package nvidia-docker2，解决方法是完整执行完第 1 步；

3--测试

        重启 docker：

1.  
   sudo systemctl daemon-reload
2.  
   sudo systemctl restart docker

        查看显卡信息（执行这步时会自动下载对应的镜像文件）：

docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi

这篇好文章是转载于：学新通技术网