TensorFlow-GPU安装配置(win10+tensorflow1.6+CUDA9.0+cudnn7.0+python3.6+Visual Studio2013)

安装步骤:

TensorFlow官网

tensorflow一般只能装在python3上,CUDA9.0搭配cudnn7.0,CUDA8.0搭配cudnn6.0

查看对应要安装的环境版本(因为会不断更新,所以一定要看清对应的版本,如下图),按照要求一步步将所有的软件下载好,并且在命令行pip装tensorflow-gpu

 

NVIDIA驱动程序下载

CUDA安装指南 for Microsoft Windows

CUDA工具包下载

cuDNN安装指南 for Microsoft Windows

cuDNN下载

 

测试是否装好:

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即可。

 

 

CUDA快速入门指南

TensorFlow小例子和简单的入门学习

 

 

 

 


 

posted @ 2018-03-23 20:48  tuji_sjp  阅读(407)  评论(0编辑  收藏  举报