ubuntu 16.04 anaconda2 下安装caffe2

主要按照官网上的提示：https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=prebuilt

首先是安装cuda，cudnn，nccl

cuda和cudnn教程较多

--------------------------------------------------------------------------------------------------------------------------------------------------------------

cuda教程：https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

----------------------------------------------------------------------------------------------------------------------------------------------------

cudnn教程：https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#overview

2.3. Installing cuDNN on Linux

The following steps describe how to build a cuDNN dependent program. Choose the installation method that meets your environment needs. For example, the tar file installation applies to all Linux platforms. The debian installation package applies to Ubuntu 14.04 and 16.04.

In the following sections:

• your CUDA directory path is referred to as /usr/local/cuda/
• your cuDNN download path is referred to as

 Installing from a Tar File

1. Navigate to your directory containing the cuDNN Tar file.
2. Unzip the cuDNN package.

   $ tar -xzvf cudnn-9.0-linux-x64-v7.tgz

3. Copy the following files into the CUDA Toolkit directory.

   $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include
   $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
   $ sudo chmod a+r /usr/local/cuda/include/cudnn.h
   /usr/local/cuda/lib64/libcudnn*

----------------------------------------------------------------------------------------------------------------------------------------------------

nccl教程参考官网:https://docs.nvidia.com/deeplearning/sdk/nccl-install-guide/index.html

 Ubuntu 14.04 LTS And Ubuntu 16.04 LTS

Installing NCCL on Ubuntu requires you to first add a repository to the APT system containing the NCCL packages, then installing the NCCL packages through APT. There are two repositories available; a local repository and a network repository. Choosing the later is recommended to easily retrieve upgrades when newer versions are posted.

1. Install the repository.
   • For the local NCCL repository:

     sudo dpkg -i nccl-repo-.deb

   • For the network repository:

     sudo dpkg -i nvidia-machine-learning-repo-.deb

2. Update the APT database:

   sudo apt update

3. Install the libnccl2 package with APT. Additionally, if you need to compile applications with NCCL, you can install the libnccl-dev package as well:
   Note: If you are using the network repository, the following command will upgrade CUDA to the latest version.

   sudo apt install libnccl2 libnccl-dev

   If you prefer to keep an older version of CUDA, specify a specific version, for example:

   sudo apt-get install libnccl2=2.0.0-1+cuda8.0 libnccl-dev=2.0.0-1+cuda8.0

   Refer to the download page for exact package versions.

---------------------------------------------------------------------------------------------------------------------------------------------------

接下来开始安装

If you do not have a GPU:

conda install -c caffe2 caffe2

For GPU support you will need CUDA, CuDNN, and NCCL. These must be installed from Nvidia’s website.

For Caffe2 with CUDA 9 and CuDNN 7 support:

conda install -c caffe2 caffe2-cuda9.0-cudnn7

For Caffe2 with CUDA 8 and CuDNN 7 support:

conda install -c caffe2 caffe2-cuda8.0-cudnn7

按照官网上的建议：

安装caffe2之前最好新建一个anaconda环境

参考网页：https://conda.io/docs/user-guide/tasks/manage-environments.html

Use the Terminal or an Anaconda Prompt for the following steps.

    To create an environment:

    conda create --name myenv

    NOTE: Replace myenv with the environment name.

    When conda asks you to proceed, type y:

    proceed ([y]/n)?

This creates the myenv environment in /envs/. This environment uses the same version of Python that you are currently using, because you did not specify a version.

完成后，进入打开caffe2的方法是：

首先激活caffe2的环境

比如我的：

然后进入python：

最后import caffe2以测试是否成功