conda虚拟环境中安装CUDA、cuDNN、TensorFlow1.11

安装完显卡驱动和anaconda之后，可以直接在虚拟环境中安装CUDA和cuDNN。我这里在复现一个深度学习网络，使用的CUDA9.0版本和cuDNN7.3.1以及TensorFlow1.11。

创建虚拟环境：

 conda create -n Randlanet python=3.5

 激活虚拟环境：

conda activate Randlanet

接下来CUDA和cuDNN全部在虚拟环境中安装。

1.CUDA安装

首先使用conda命令查看conda中有哪些版本的CUDA可以安装：

conda search cudatoolkit --info

从列出的版本中选择一个版本安装。比如安装CUDA9.0：

 

conda install cudatoolkit=9.0

2.cuDNN安装

使用conda命令查看conda中有哪些cudnn版本可以安装：

conda search cudnn --info

 从中选择一个版本安装，比如：

在虚拟环境中安装cuDNN7.3.1:

conda install cudnn=7.3.1

 至此，CUDA和cuDNN安装完成。

3.安装TensorFlow1.11

pip install tensorflow-gpu==1.11 -i https://pypi.tuna.tsinghua.edu.cn/simple  --timeout=120

提示python版本不对，更新pip即可解决问题。

 重新安装：

 至此，安装完毕。