libtorch_cuda_cu.so: cannot open shared object file

1.问题描述:

from mmdet.apis import inference_detector, init_detecto

运行时报错:

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

(1)环境中libtorch_cuda_cu.so文件:libtorch_cuda_cu.so: cannot open shared object file

2.原因和解决方法:

mmcv的版本和torch版本不一致
[mmcv官网: PyTorch 和 CUDA 版本要求]
[mmcv官网: 其他版本PyTorch ] 找对应版本torch 并重装

conda install -c pytorch pytorch=1.3.0 cudatoolkit=10.1 torchvision -y

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2023年9月7日
下一篇 2023年9月7日

相关推荐