TensorFlow-GPU安装配置(win10+tensorflow1.6+CUDA9.0+cudnn7.0+python3.6+Visual Studio2013)
安装步骤:
tensorflow一般只能装在python3上,CUDA9.0搭配cudnn7.0,CUDA8.0搭配cudnn6.0
查看对应要安装的环境版本(因为会不断更新,所以一定要看清对应的版本,如下图),按照要求一步步将所有的软件下载好,并且在命令行pip装tensorflow-gpu
CUDA安装指南 for Microsoft Windows
cuDNN安装指南 for Microsoft Windows
测试是否装好:
1、在python中 import tensorflow 不报错
2、系统环境变量新增“CUDA_PATH”和“CUDA_PATH_版本”,path变量新增CUDA工具包的路径
3、运行两个CUDA自带的实例,C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\1_Utilities\deviceQuery中的deviceQuery_vs2013.sln,C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\1_Utilities\bandwidthTest中的bandwidthTest_vs2013.sln,控制台都能输出PASS
全部装好后,TensorFlow-GPU版本就可以正常使用了。但是出现了一个小错误,是因为numpy的版本之前可能更新错了版本,导致很多库如skimage、sklearn等都无法正常使用,出现“DLL load failed: 找不到指定的模块”的错误。后来找到一篇超强的安装tensorflow错误总结博客,终于解决了问题。分享给大家:https://blog.csdn.net/u010099080/article/details/53418159#reply 几乎所有安装过程时可能出现的问题都会在上边找到,如果实在没有可以马上问博主,博主超nice,一直到现在都会及时回复。
解决方法:重新下载numpy+mkl的whl,重装numpy即可。