簡單實現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即可