您的位置:首页 > 其它

win10 TensorFlow GPU环境搭建

2018-04-03 11:27 197 查看

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。

又去找教程,于是有了第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

8d4b
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: