赞
踩
安装完显卡驱动和anaconda之后,可以直接在虚拟环境中安装CUDA和cuDNN。我这里在复现一个深度学习网络,使用的CUDA9.0版本和cuDNN7.3.1以及TensorFlow1.11。
conda create -n Randlanet python=3.5
激活虚拟环境:
conda activate Randlanet
接下来CUDA和cuDNN全部在虚拟环境中安装。
首先使用conda命令查看conda中有哪些版本的CUDA可以安装:
conda search cudatoolkit --info
从列出的版本中选择一个版本安装。比如安装CUDA9.0:
conda install cudatoolkit=9.0
使用conda命令查看conda中有哪些cudnn版本可以安装:
conda search cudnn --info
从中选择一个版本安装,比如:
在虚拟环境中安装cuDNN7.3.1:
conda install cudnn=7.3.1
至此,CUDA和cuDNN安装完成。
pip install tensorflow-gpu==1.11 -i https://pypi.tuna.tsinghua.edu.cn/simple --timeout=120
提示python版本不对,更新pip即可解决问题。
重新安装:
至此,安装完毕。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。