win10环境下tensorflow-gpu安装

ayew 2017-11-25

dll

 

 

 

 

踩了很多坑,浪费了一下午时间,在此记录一下。

pip install tensorflow-gpu

 

然后下载安装cuda8.0

https://developer.nvidia.com/cuda-downloads

注意:只能用8.0

我用9.0下载了好久,程序提示我装8.0

然后:下载cudnn

https://developer.nvidia.com/rdp/cudnn-download

 

要注册才能下载,这又是一个坑,选定cudnn6.0版本

其中,cudnn的安装方法是将cudnn下面的三个文件夹(bin,include, lib)复制到cuda 8.0文件夹。

装完以后运行,出现以下错误:

ImportError: No module named '_pywrap_tensorflow_internal'

ImportError: DLL load failed: The specified module could not be found.

解决的办法是把cudnn64_6.dll改名为cudnn64_5.dll。

然后

import tensorflow

没有报错就成功了。

 

登录 后评论
下一篇
云栖号资讯小编
408人浏览
2020-06-01
相关推荐
Win10安装TensorFlow1.9-GPU版本
1613人浏览
2018-08-13 19:27:00
Ubuntu18.04安装Tensorflow-gpu
2268人浏览
2018-06-06 04:13:00
关于TensorFlow的安装和心得
498人浏览
2018-01-30 00:14:00
QDS04 TensorFlow
996人浏览
2019-02-21 18:57:16
TensorFlow——快速安装
1479人浏览
2018-04-19 19:34:03
keras 使用笔记
1109人浏览
2017-06-25 15:07:00
深度学习必备包
708人浏览
2018-01-10 19:19:00
0
0
0
733