docker for caffe

安装：

＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝＝

Ubuntu 16.04

docker 17.03

nvidia driver安装好

 

安装nvidia-docker：

# Install nvidia-docker and nvidia-docker-plugin
wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb

# Test nvidia-smi
nvidia-docker run --rm nvidia/cuda nvidia-smi

 

关闭docker服务：

systemctl  stop docker.service

启动nvidia-docker服务

systemctl start nvidia-docker.service

 

安装bvlc/caffe

nvidia-docker run -ti bvlc/caffe:gpu caffe --version

 

加载一个host目录