使用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版本.