search

FishYuLi / BalancedGroupSoftmax

CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.
Apache License 2.0
352 stars 64 forks source link

ImportError: libtorch_cpu.so: cannot open shared object file: No such file or directory #7

Open U-Help opened 4 years ago

U-Help commented 4 years ago

Traceback (most recent call last): File "tools/train.py", line 9, in from mmdet.apis import (get_root_logger, init_dist, set_random_seed, File "/home/qth/BalancedGroupSoftmax/mmdet/apis/init.py", line 2, in from .inference import (inference_detector, init_detector, show_result, File "/home/qth/BalancedGroupSoftmax/mmdet/apis/inference.py", line 11, in from mmdet.core import get_classes File "/home/qth/BalancedGroupSoftmax/mmdet/core/init.py", line 6, in from .post_processing import * # noqa: F401, F403 File "/home/qth/BalancedGroupSoftmax/mmdet/core/post_processing/init.py", line 1, in from .bbox_nms import multiclass_nms File "/home/qth/BalancedGroupSoftmax/mmdet/core/post_processing/bbox_nms.py", line 3, in from mmdet.ops.nms import nms_wrapper File "/home/qth/BalancedGroupSoftmax/mmdet/ops/init.py", line 7, in from .nms import nms, soft_nms File "/home/qth/BalancedGroupSoftmax/mmdet/ops/nms/init.py", line 1, in from .nms_wrapper import nms, soft_nms File "/home/qth/BalancedGroupSoftmax/mmdet/ops/nms/nms_wrapper.py", line 4, in from . import nms_cpu, nms_cuda ImportError: libtorch_cpu.so: cannot open shared object file: No such file or directory

When I follow the steps of README.md and use "CUDA_VISIBLE_DEVICES=1 python tools/train.py configs/bags/gs_faster_rcnn_r50_fpn_1x_lvis_with0_bg8.py" to train the model, I meet the above problems, can you tell me how to solve it?