环境信息:
显卡 2080ti
cudatoolkit
(version 11.0.221
)
cuDNN
(version 8.0.4
tensorflow-gpu
(version 2.4.1
).
问题:
使用编译的tensorflow.dll预测时,cuda和cudnn正常加载,但出现错误无法正常使用GPU。错误信息为:
2021-07-16 19:17:09.091737: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library cudnn64_8.dll
2021-07-16 19:17:09.712092: I tensorflow/stream_executor/cuda/cuda_dnn.cc:334] Loaded cuDNN version 8,100
2021-07-16 19:17:10.409973: E tensorflow/core/platform/windows/subprocess.cc:283] Call to CreateProcess failed. Error code: 2
2021-07-16 19:17:10.415635: W tensorflow/stream_executor/gpu/asm_compiler.cc:56] Couldn't invoke ptxas.exe --version
2021-07-16 19:17:10.424879: E tensorflow/core/platform/windows/subprocess.cc:283] Call to CreateProcess failed. Error code: 2
2021-07-16 19:17:10.431698: W tensorflow/stream_executor/gpu/redzone_allocator.cc:314] Internal: Failed to launch ptxas
Relying on driver to perform ptx compilation.
Modify $PATH to customize ptxas location.
注意到错误是:Couldn't invoke ptxas.exe --version
这是由于软件环境和显卡不匹配导致的,在2080ti这类偏旧的显卡推荐使用cuda10+cudnn7系列的环境,在3090,a100这类较新的分别推荐至少cuda11+cudnn8和使用cuda11.2+cudnn8.1系列的环境。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)