Ubuntu下使用conda在虛擬環境中安裝CUDA、CUDNN及Tensorflow

在Ubuntu中安裝Anaconda可以根據 這篇部落格,寫的十分詳細。
裝好之後，可以通過通過命令安裝版本為X.X(如2.7，3.6）的虛擬環境：

conda create -n your_env_name python=X.X

其中(your_env_name)是你所建立的虛擬環境的名字。
然後 使用下方命令啟用你剛才建立的環境

source activate your_env_name

啟用了環境之後，我們可以看到命令列之前多了一個括號，括號中包含了你的環境名字，這就說明啟用成功。
在啟用的環境中使用如下命令安裝CUDA：

conda install cudatoolkit=8.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

我裝CUDA的同時，自動裝上了CUDNN。
如果你沒有裝上，也可以使用下方命令安裝CUDNN：

conda install cudnn=7.0.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

其中CUDA的版本可以自己更改，我改成了9.2.0，CUDNN應該也可以改，不過我沒有嘗試。
這篇文章還展示了Tensorflow不同版本要求與CUDA及CUDNN版本對應關係，一般我們都是裝好了tensorflow-gpu，然後需要對應版本的CUDA及CUDNN，可以讓大家參考一下。
仍然是在啟用的環境中安裝tensorflow，比如我的CUDA是9.2.0，我的CUDNN是7.2.1，那麼對應的Tensorflow就可以安裝1.5.0，如下方命令所示：

pip install tensorflow-gpu==1.5.0

以上所安裝的Python、CUDA、CUDNN以及Tensorflow都是在虛擬環境中安裝的，下次想要使用該環境時或者在該環境安裝其他庫包檔案，需要重新進入虛擬環境。與之前的程式碼一樣，如下：

source activate your_env_name`

查詢當前環境下的庫的版本號可使用以下命令：

conda list cudnn
conda list cuda