当前位置: 首页 > 工具软件 > Caffe2 > 使用案例 >

caffe2安装篇(三)通过docker安装

督劲
2023-12-01

用普通的安装方式走了不少弯路,感觉还是用docker方便:

参考的是https://hub.docker.com/r/caffe2ai/caffe2/

Latest

docker pull caffe2ai/caffe2
Comes with GPU support, CUDA 8.0, cuDNN 7, all options, and tutorial files. Uses Caffe2 v0.8.1.

GPU images (for use with nvidia-docker)

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.

  • Caffe2 v0.8.1, CUDA 8, cuDNN 7: docker pull caffe2ai/caffe2:c2v0.8.1.cuda8.cudnn7.ubuntu16.04
  • Caffe2 v0.8.0, CUDA 8, cuDNN 6: docker pull caffe2ai/c2v0.8.0.cuda8.cudnn6.ubuntu16.04
  • Caffe2 v0.7.0, CUDA 8, cuDNN 6: docker pull caffe2ai/c2.cuda8.cudnn6.ubuntu16.04
  • Caffe2 v0.7.0, CUDA 8, cuDNN 5: docker pull caffe2ai/c2.cuda8.cudnn5.ubuntu16.04

CPU Images

 类似资料: