在Ubuntu16.04 CUDA9.0 cuDNN8.0的环境下安装caffe2
本博客比较简单,cuda9.0 cudnn8.0部分请看上一篇博客,其中详细讲了:
- 如何安装驱动
- 安装cuda
- 安装cudnn
- 安装tensorflow
本教程主要参考来自Caffe2官方说明文档
For GPU support you will need CUDA, CuDNN, and NCCL. These must be installed from Nvidia’s website.
选择ubuntu,Pre-built Binaries,其中有一个安装方式:
For Caffe2 with CUDA 9 and CuDNN 7 support:
conda install -c caffe2 caffe2-cuda9.0-cudnn7
但是我们要先将NCCL安装,选择NCCL 2.2.13 for Ubuntu 16.04 and CUDA 9
,下载下来的文件是deb格式的
我的全称为:nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.deb
开始安装:
sudo dpkg -i nccl-repo-ubuntu1604-2.2.13-ga-cuda9.0_1-1_amd64.deb
最后在你的conda环境下运行conda install -c caffe2 caffe2-cuda9.0-cudnn7
即可