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文件:

不存在

red {不存在}

不存在

2.原因和解决方法:

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

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

在这里插入图片描述
在这里插入图片描述

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇

)">
下一篇>>