如何在windows上安装tensorflow+gpu

如何在windows上安装tensorflow+gpu

  1. windows安装tensorlfow一定是要gpu版本
  2. 安装流程先安装cuda+cudnn
  3. 然后将cudnn里面文件考到装好的cuda中,网上很多教程
  4. Anaconda下载安装
  5. 官网的中的安装tensorflow步骤
    参考步骤一
    参考步骤二
    内容大同小异,一般都能安装成功

  6. 最后import tensorflow as tf会报错,类似如下:

Traceback (most recent call last): 
File “D:\Python\Python35\lib\site-packages\tensorflow\python\pywrap_tensorflow_internal.py”, line 18, in swig_import_helper 
return importlib.import_module(mname) 
File “D:\Python\Python35\lib\importlib__init__.py”, line 126, in import_module 
return _bootstrap._gcd_import(name[level:], package, level) 
File “frozen importlib._bootstrap”, line 986, in _gcd_import 
File “frozen importlib._bootstrap”, line 969, in _find_and_load 
File “frozen importlib._bootstrap”, line 958, in _find_and_load_unlocked 
File “frozen importlib._bootstrap”, line 666, in _load_unlocked 
File “frozen importlib._bootstrap”, line 577, in module_from_spec 
File “frozen importlib._bootstrap_external”, line 906, in create_module 
File “frozen importlib._bootstrap”, line 222, in _call_with_frames_removed 
ImportError: DLL load failed: 找不到指定的模块。

During handling of the above exception, another exception occurred:

找不到指定模块,出现这个问题一般都是 cuda+cudnn版本和tensorflow版本不一致。
一般在安装tensoflow,我们习惯会直接pip install tensorflow-gpu这时会默认安装tensorflow官网给出的最新版本,现在是1.4。从1.3版本以后,tensoflow就不再支持cudnn的v5.1版本,所以最好是下载二进制文件,用pip install+二进制文件的安装方式来安装cudnn的v5.1版本的tensorflow,或者下载cudnn的v6版本,安装tensorflow1.3 1.4版本。

例如我用的是cuda V8+cudnn V5.1版本的,在安装后anaconda后,从tensorflow1.2whl文件网站下载tensorflow-1.2.1-cp35-cp35m-win_amd64.whl (md5)文件,再pip install tensorflow-1.2.1-cp35-cp35m-win_amd64.whl就可以安装成功了。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值