在windows10使用tensorflow1.0.0和python3.5的环境下,我的cuda使用的9.0。
在运行:
1 |
>>> import tensorflow as tf |
出现以下错误
1 2 3 4 |
108] Couldn't open CUDA library cublas64_80.dll 102] Couldn't open CUDA library cudnn64_5.dll 108] Couldn't open CUDA library cufft64_80.dll 108] Couldn't open CUDA library curand64_80.dll |
这是因为tensorflow1.0.0只能支持到80,所以他会找***_80.dll,cudnn64_5.dll,但这些dll都不存在,怎么办呢?
其实很简单:
1.来到CUDA\v9.0\bin
2.复制cublas64_90.dll cudnn64_7.dll cufft64_90.dll curand64_90.dll 将其分别复制成新的 cublas64_80.dll cudnn64_5.dll cufft64_80.dll curand64_80.dll
重启命令行工具
重新输入
1 |
>>> import tensorflow as tf |
1 2 3 4 |
108] successfully opened CUDA library cublas64_80.dll 102] successfully opened CUDA library cudnn64_5.dll 108] successfully opened CUDA library cufft64_80.dll 108] successfully opened CUDA library curand64_80.dll |
大功告成!!!!
如果遇到更多tensorflow的gpu问题,请看:http://www.newlifeclan.com/ai/archives/232