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

环境信息:

显卡 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系列的环境。

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值