简单实现Ubuntu16.04 + caffe2 + CUDA9.0 + cuDNN8.0

在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​​即可

代码改变世界