ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

动态链接库问题

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的完整路径)

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值