NVIDIA Docker 用 GPU 运行 caffe

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

其中：

1. “-it”：表示用交互模式运行docker
2. “–gpus”：指定GPU
3. “-v”：将外部路径（冒号前面的）挂载在docker中（冒号后面的），这样可以在docker中访问外部文件，方便docker中的文件读写。这里将外部路径/media/ricky/5AA031C1A031A483映射为容器内的Disk目录。
4. “nvidia/caffe”：为docker hub上的NVIDIA官方caffe镜像：https://hub.docker.com/r/nvidia/cuda。换成你所需要的镜像名字和版本。
5. “/bin/bash”：表示进入bash命令行