CuDNN是一个专门针对Deep Learning框架设计的GPU计算加速方案,目前支持的DL库包括Caffe,ConvNet, Torch7等。需要的朋友欢迎下载!
CuDNN可以在官网免费获得,注册帐号后即可下载。官网没有找到安装说明,基本原理是把lib文件加入到系统能找到的lib文件夹里, 把头文件加到系统能找到的include文件夹里就可以。
pip install tensorflow-gpu后
Anaconda prompt中输入:import tensorflow时,报错:ImportError,需要用到这个库文件,分别copy至CUDA\v8.0 中三个目录下:bin,include,lib
具体错误一般如下:
File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 18, in swig_import_helper
return importlib.import_module(mname)
File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "
File "
File "
File "
File "
File "
File "
ImportError: DLL load failed: The specified module could not be found.
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "C:\Anaconda3\lib\site-packages\tensorflow\python\__init__.py", line 66, in
from tensorflow.python import pywrap_tensorflow
File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 21, in
_pywrap_tensorflow = swig_import_helper()
File "C:\Anaconda3\lib\site-packages\tensorflow\python\pywrap_tensorflow.py", line 20, in swig_import_helper
return importlib.import_module('_pywrap_tensorflow')
File "C:\Anaconda3\lib\importlib\__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
ImportError: No module named '_pywrap_tensorflow'
During handling of the above exception, another exception occurred:
。。。。。
下载体验
下载仅供下载体验和测试学习,不得商用和正当使用。