您的位置:首页 > 其它

如何在windows上安装tensorflow+gpu

2017-11-01 10:11 337 查看

如何在windows上安装tensorflow+gpu

windows安装tensorlfow一定是要gpu版本

安装流程先安装cuda+cudnn

然后将cudnn里面文件考到装好的cuda中,网上很多教程

Anaconda下载安装

官网的中的安装tensorflow步骤

参考步骤一

参考步骤二

内容大同小异,一般都能安装成功

最后
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
就可以安装成功了。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  windows tensorflow cuda