tensorflow无法正常使用显卡,错误Couldn‘t invoke ptxas.exe --version

2023-05-16

环境信息:

显卡 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(使用前将#替换为@)

tensorflow无法正常使用显卡,错误Couldn‘t invoke ptxas.exe --version 的相关文章

随机推荐