当前位置:   article > 正文

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

conda安装cuda

安装完显卡驱动和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即可解决问题。

 重新安装:

 至此,安装完毕。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/笔触狂放9/article/detail/398011
推荐阅读
相关标签
  

闽ICP备14008679号