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即可解决问题。

 重新安装:

 至此,安装完毕。

猜你喜欢

转载自blog.csdn.net/m0_62648611/article/details/131731563