win10 TensorFlow GPU环境搭建

win10+cuda9.1+cudnn7.1.1+tf1.6+py3.6.5+vs2015

这篇文章只是记录搭建的流程以及一些坑的解决方法,并不是详细的教程,也没有截图什么的

我电脑上本身是装了vs2015的,省了一些功夫。

1.安装cuda9.1

cuda官网上下载cuda9.1。
local安装包总是下载到一般就停了,暂停一下继续下载之后就变成了403.html
于是换了network安装包。

在安装之前需要更新一下显卡驱动,否则安装可能会失败

启动安装包,我选了精简安装。等它装好。。
安装时他会自动帮你添加环境变量,去cmd里输入nvcc -V检验一下。

之后再编译一下cuda示例程序(在c:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.1)。如果提示是缺少:
“d3dx9.h”、”d3dx10.h”、”d3dx11.h”头文件 ,安装一下DirectX SDK。
最后还是有一个项目失败了,,不管他,貌似不要紧。

然后在cmd中运行c:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.1\bin\win64\Release\deviceQuery.exe这个文件检验一下。

2.安装cudnn7.1.1

cudnn官网上下载,需要注册一个账号。
不出所料,下到一半就停了(而且都是在98%停的)。
然后在csdn下了一个。

解压后把里面3个文件夹覆盖到cuda安装目录下。

3.安装TensorFlow

(下面是坑,先不要踩)
pip install,,但是官方的whl不支持cuda9.1,就算安装成功,import的时候也会出错(错误貌似是找不到’cudart64_90.dll’)。

(下面是解决方法)
一个好人的github上找到对应的whl,这里用的是1.6.0\py36\GPU\cuda91cudnn71avx2里的whl。
打开cmd -> 切换到你保存whl的目录 -> pip install tensorflow_gpu-1.6.0-cp36-cp36m-win_amd64.whl-> 成功出错。
记得好像是说当前平台不支持这个whl。

扫描二维码关注公众号,回复: 1820115 查看本文章

又去找教程,于是有了第4步

4.安装anaconda

去官网上下载,速度很慢,可以到清华开源软件镜像站下载

安装完后打开cmd,输入conda create -n tensorflow python=3.6
让你选([y]/n)输入y
结束后再输入activate tensorflow
然后就切换到你保存tensorflow的whl的目录,输入pip install tensorflow_gpu-1.6.0-cp36-cp36m-win_amd64.whl等安装完成

(conda create为tensorflow创建了一个环境,详细的操作可以看这篇博客

参考

https://blog.csdn.net/qq_36556893/article/details/79433298
https://www.jianshu.com/p/c245d46d43f0
https://blog.csdn.net/vcvycy/article/details/79298703

猜你喜欢

转载自blog.csdn.net/sgafpzys/article/details/79798375