官方说随着docker19.03的发布,不赞成使用nvidia-docker2软件包,因为Docker运行时中现在已将NVIDIA GPU作为设备本地支持。但是折腾了几天,安装了docker19.03,也用不起NVIDIA GPU,所以这里还是退而求其次,安装低版本的docker,并且安装nvidia-docker2,先用起来再说,后续再研究。
环境:
Ubuntu 16.04 LTS
NVIDIA driver 390.87
Docker 18.09.7
nvidia-docker2 2.0.3
nvidia-container-runtime 2.0.0
先卸载nvidia-docker
sudo docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f sudo apt-get purge nvidia-docker
选择和已安装的docker配套的版本。
如果没有镜像会自动下载。
说明已经调用到GPU了。
安装过程遇到的两个错误请参考链接:
1. 【Docker】nvdia-docker2 docker-ce (= 5 18.09.7~3-0~ubuntu-xenial)but 5 19.03.4~3-0~ubuntu-xenial is to
2. 【Docker】nvidia-container-runtime Depends nvidia-container-runtime-hook