用普通的安装方式走了不少弯路,感觉还是用docker方便:
参考的是https://hub.docker.com/r/caffe2ai/caffe2/
docker pull caffe2ai/caffe2
Comes with GPU support, CUDA 8.0, cuDNN 7, all options, and tutorial files. Uses Caffe2 v0.8.1.
All GPU builds come with Ubuntu 16.04 and fully loaded with optional dependencies like OpenCV. Minimal builds and Ubuntu 14.04 can be found in the CPU section.
docker pull caffe2ai/caffe2:c2v0.8.1.cuda8.cudnn7.ubuntu16.04
docker pull caffe2ai/c2v0.8.0.cuda8.cudnn6.ubuntu16.04
docker pull caffe2ai/c2.cuda8.cudnn6.ubuntu16.04
docker pull caffe2ai/c2.cuda8.cudnn5.ubuntu16.04