动态链接库问题
ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory
报错原因是:环境的torch版本低 torch/lib 中不包含libtorch_cuda_cu.so这一动态链接库(.so在linux中相当于windows中的.dll)
很多人推荐更新torch 但这可能会导致与本机cuda版本不匹配
其实直接补充下载libtorch就行了
Linux:[LibTorch & Linux] 各版本 LibTorch 下载-CSDN博客
Windows:[LibTorch & Win] 各版本 LibTorch 下载_linux下libtorch的gpu版本下载-CSDN博客
可以参考以上两个博客找匹配自己cuda和torch版本的libtorch并下载
下载好之后libtorch_cuda_cu.so等动态链接库应该在libtorch/lib中
之后使用
export LD_LIBRARY_PATH= :$LD_LIBRARY_PATH
将libtorch/lib加入系统环境变量即可(空白的部分是到libtorch/lib的完整路径)