本文参考了其他内容。
- 安装nvidia驱动
注意，安装cuda自带的驱动，经常装不上。所以强烈建议单独安装驱动。
具体步骤参考：《 Ubuntu16.04+NVIDIA显卡驱动安装步骤》
当然，软件版本可以用最新的。
- 安装cuda
提示安装驱动时，选择no。
其他yes、默认即可。
- 安装docker
# step 1: 安装必要的一些系统工具 sudo apt-get update sudo apt-get -y install apt-transport-https ca-certificates curl software-properties-common # step 2: 安装GPG证书 curl -fsSL http://mirrors.aliyun.com/docker-ce/linux/ubuntu/gpg | sudo apt-key add - # Step 3: 写入软件源信息 sudo add-apt-repository "deb [arch=amd64] http://mirrors.aliyun.com/docker-ce/linux/ubuntu $(lsb_release -cs) stable" # Step 4: 更新并安装 Docker-CE sudo apt-get -y update sudo apt-get -y install docker-ce
- 安装nvidia-docker2
# 清理以前的。If you have nvidia-docker 1.0 installed: we need to remove it and all existing GPU containers 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 -y nvidia-docker sudo apt autoremove # 执行命令。Add the package repositories # command 1 curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \ sudo apt-key add - # command 2 distribution=$(. /etc/os-release;echo $ID$VERSION_ID) # command 3 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 # 正式安装。Install nvidia-docker2 and reload the Docker daemon configuration sudo apt-get install -y nvidia-docker2 sudo pkill -SIGHUP dockerd # 测试一下。 Test nvidia-smi with the latest official CUDA image sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi
- 出错的解决办法
《解决办法：nvidia-docker2 : 依赖: docker-ce (= 5:18.09.0~3-0~ubuntu-xenial)》
