docker用gpu的参数_docker使用GPU

1.物理机安装显卡驱动

2.安装nvidia-docker

wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.0-rc.3/nvidia-docker_1.0.0.rc.3-1_amd64.deb

sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb

3.下载image

nvidia-docker pull nvidia/cuda:6.5

4.启动

nvidia-docker run --rm nvidia/cuda:6.5 nvidia-smi

docker: Error response from daemon: create nvidia_driver_352.93: create nvidia_driver_352.93: Error looking up volume plugin nvidia-docker: plugin not found.

出错的话，先运行 nvidia-docker-plugin

阿里云加速

curl -sSL http://acs-public-mirror.oss-cn-hangzhou.aliyuncs.com/docker-engine/internet | sh -

echo "DOCKER_OPTS=\"\$DOCKER_OPTS --registry-mirror=https://aimk.mirror.aliyuncs.com\"" | sudo tee -a /etc/default/docker

sudo service docker restart

https://github.com/NVIDIA/nvidia-modprobe 下载编译 安装