TensorFlow-gpu 安装各疑难杂症解答
TensorFlow-gpu 安装各疑难杂症解答
高级安装教程,有问题联系博主
该教程适用于想自己安装各组件部分的人群,想偷懒的人请转道使用Anaconda安装,贴上别人的教程。
https://www.cnblogs.com/HongjianChen/p/8385547.html
1.请确保自己的python版本为3.6!!!
python3.7会出现一系列问题,所以在安装前请更换好python版本。
- 下载链接 ,https://www.python.org/downloads/release/python-366/;
- python安装可自行寻找教程,推荐高级安装,指定安装文件夹,建议将能勾选的全勾上,特别是添加到path那个选项。
2.安装pycharm(可选)
pycharm对于各种包的管理很方便,运行py程序也很方便。
- 下载链接 ,http://www.jetbrains.com/pycharm/
- pycharm安装可自行寻找教程,这里不再赘述。
下载安装CUDA
下载前请对照tensorflow版本对应表
https://blog.csdn.net/omodao1/article/details/83241074
一般是安装CUDA9.0,(不建议安装9.2,会出错)
这是让你选择安装CUDA所使用的临时文件夹 !!!安装完成后该文件夹会自动删除。
切记勿和下一步选择安装文件夹使用同一文件夹,否则安装后文件夹会被删除!
一步一步往下走就行了,勾选建议全勾。
下载CUDNN
对应CUDA版本下载!
下载解压后
将这三个文件夹复制到CUDA目录下
划重点!
将bin,lib/×64,和cuda目录添加到环境变量中!
安装TensorFlow-gpu
未安装pycharm的人,请使用cmd命令
pip install tensorflow-gpu (默认安装支持的最新版本,途中会安装其他需要的包,网速好的话大概10分钟)
嫌速度慢的可以使用镜像安装,教程自行查找
到这里基本算是大功告成了!接下来就是解决疑难杂症的时候!
一般情况下到这里还会出现各种各样的问题
- 缺少cudart.dll ,cuda未能正确导入,请检查tensorflow,cuda,cudnn的版本是否对应
- dll load faied,确保自己电脑上安装了microsoft visual c++2015 redistributable或者2017
- 大多数情况都是这两个问题,其他问题都在上述步骤中避免了,如果按照教程做完后还有其他问题可以联系博主。