docker中使用gpu

安装nvidia-docker2

依次输入下面命令：

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey sudo apt-key add -

distribution=(./etc/os−release;echoID$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 nvidia-docker2

sudo pkill -SIGHUP dockerd

修改/etc/docker/daemon.json

{ “default-runtime”: “nvidia”, “runtimes”: { “nvidia”: { “path”: “/usr/bin/nvidia-container-runtime”, “runtimeArgs”: [], “registry-mirrors”: [“https://gemfield.mirror.aliyuncs.com”] } } }

sudo systemctl restart docker

sudo docker run –runtime nvidia nvidia/cuda:9.0-base nvidia-smi

注意：Docker 19.03以上版本请使用NVIDIA Container Toolkit 工具包