torch报错缺少libcudnn_ops_infer.so.8解决方法

搬运贴,其他网站看到的比较有用记录一下

  • 问题及解决方案
  • 解决方案

问题及解决方案

可能是cudnn出现问题了。可以根据这篇文章的命令验证一下torch.backends.cudnn.version()

解决方案

参考知乎https://zhuanlan.zhihu.com/p/333362435,第3.4节和第4节的的教程进行操作,解压cudnn文件后vim ~/.bashrc或者vim ~/.zshrc修改路径

export CUDA_HOME=/xxx/cudnn-linux-xxx#改成自己的解压路径
export LD_LIBRARY_PATH=${CUDA_HOME}/lib:$LD_LIBRARY_PATH

你可能感兴趣的:(笔记,linux,python,pytorch)