本文共 1210 字，大约阅读时间需要 4 分钟。

第一步，安装docker 

sudo apt-get remove docker docker-engine docker.io containerd runc sudo apt-get update sudo apt-get install \    apt-transport-https \    ca-certificates \    curl \    gnupg-agent \    software-properties-common curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add - sudo add-apt-repository \   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \   $(lsb_release -cs) \   stable"

第二步，安装 Nvidia-Docker  

sudo apt-get update sudo apt-get install docker-ce docker-ce-cli containerd.io apt-cache madison docker-ce 选一个  for example, 5:18.09.1~3-0~ubuntu-xenialsudo apt-get install docker-ce=
   
     docker-ce-cli=
    
      containerd.io sudo docker run hello-world#运行通过就OKdocker version#有版本提示就OK
    
   

 第三步  安装Caffe2

docker pull caffe2ai/caffe2 # to testnvidia-docker run -it caffe2ai/caffe2:latest python -m caffe2.python.operator_test.relu_op_test # to interactnvidia-docker run -it caffe2ai/caffe2:latest /bin/bash

 第四步 测试

python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"#返回Success就OKpython2 -c 'from caffe2.python import workspace; print(workspace.NumCudaDevices())'#返回1就OK#进入python输入from caffe2.python import workspace#不报错就OK

转载地址：http://ezczb.baihongyu.com/