使用Anaconda 創建指定cuda 版本的虛擬環境

目的:解決多cuda版本共存問題.

需要了解Tensorflow 與cuda,cudnn版本的對應關係
參考網址: https://tensorflow.google.cn/install/source

首先修改anaconda 爲清華源 參考網址: https://mirror.tuna.tsinghua.edu.cn/help/anaconda/

創建一個虛擬環境
conda create –n py3.6tf python=3.6 #創建Python3.6版本的名爲py3.6tf的虛擬環境
conda activate py3.6tf #激活環境
conda search cudatoolkit #查詢支持的cuda版本

conda search cudnn #查詢支持的cudnn版本,需要確保cuda版本一致

conda install cudatoolkit=9.0 cudnn=7.3.1

以上就完成了在anaconda虛擬環境中安裝cuda=9.0,cudnn=7.3.1的操作.這樣不同的虛擬環境可以使用不同的cuda版本.