1. 首先需要安装NVIDIA的docker:
ubuntu:
# Add the package repositories
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
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 -y nvidia-container-toolkit
sudo systemctl restart docker
CentOS:
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | sudo tee /etc/yum.repos.d/nvidia-docker.repo
sudo yum install -y nvidia-container-toolkit
sudo systemctl restart docker
参考:https://github.com/NVIDIA/nvidia-docker
2. 运行以下docker run命令即可:
sudo docker run -it --gpus all -v /media/ricky/5AA031C1A031A483:/Disk nvidia/caffe /bin/bash
其中:
- “-it”:表示用交互模式运行docker
- “–gpus”:指定GPU
- “-v”:将外部路径(冒号前面的)挂载在docker中(冒号后面的),这样可以在docker中访问外部文件,方便docker中的文件读写。这里将外部路径
/media/ricky/5AA031C1A031A483
映射为容器内的Disk
目录。 - “nvidia/caffe”:为docker hub上的NVIDIA官方caffe镜像:https://hub.docker.com/r/nvidia/cuda。换成你所需要的镜像名字和版本。
- “/bin/bash”:表示进入bash命令行