qwe3223678qwe 发表于 2017-12-6 11:14:00

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 下载编译 安装
页: [1]
查看完整版本: docker使用GPU