默认的Docker环境不支持GPU,需要安装nvidia-docker的环境。
环境:
Ubuntu 16.04 LTS
Nvidia Driver Version: 396.26
安装Docker
这里以安装docker-18.0.6-ce为例,打开https://docs.docker.com/install/ 可以看到详细教程,主要是:
卸载旧软件(没有可以忽略)
sudo apt-get remove docker docker-engine docker.io
下载deb
以下载deb包安装为例,根据自己的Ubuntu版本
去https://download.docker.com/linux/ubuntu/dists/ 选择对应自己Ubuntu发行版的源(这里我的是16.04 LTS,代号Xenial),docker-ce的安装包就是https://download.docker.com/linux/ubuntu/dists/xenial/pool/stable/amd64/ <