如何在windows上安装tensorflow+gpu
- windows安装tensorlfow一定是要gpu版本
- 安装流程先安装cuda+cudnn
- 然后将cudnn里面文件考到装好的cuda中,网上很多教程
- Anaconda下载安装
最后
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
就可以安装成功了。