最近跑模型,根据requirements.txt安好包后发现将模型导入gpu时(model.to(device))特别慢,且最终出现如下报错:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
尝试多种方法无果,想到可能是torch版本太老,与当前cuda版本不匹配,于是安装了一个较新的torch版本,问题得以解决。
若是在配环境时发现torch版本较老,可以尝试新版本的torch。
最近跑模型,根据requirements.txt安好包后发现将模型导入gpu时(model.to(device))特别慢,且最终出现如下报错:
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
尝试多种方法无果,想到可能是torch版本太老,与当前cuda版本不匹配,于是安装了一个较新的torch版本,问题得以解决。
若是在配环境时发现torch版本较老,可以尝试新版本的torch。