动态链接库libtorch_cuda_cu.so not found in training at Fine-tuning on D4LA

[1codei]

python train_VGT.py --config-file Configs/cascade/D4LA_VGT_cascade_PTM.yaml --num-gpus 8 MODEL.WEIGHTS 'VGT_pretrain_model' OUTPUT_DIR 'D4LA_output'
Traceback (most recent call last):
File "train_VGT.py", line 20, in
from detectron2.data import build_detection_train_loader
File "/home/xia/anaconda3/envs/xia-vgt/lib/python3.8/site-packages/detectron2/data/init.py", line 4, in
from .build import (
File "/home/xia/anaconda3/envs/xia-vgt/lib/python3.8/site-packages/detectron2/data/build.py", line 13, in
from detectron2.structures import BoxMode
File "/home/xia/anaconda3/envs/xia-vgt/lib/python3.8/site-packages/detectron2/structures/init.py", line 3, in
from .image_list import ImageList
File "/home/xia/anaconda3/envs/xia-vgt/lib/python3.8/site-packages/detectron2/structures/image_list.py", line 8, in
from detectron2.layers.wrappers import shapes_to_tensor
File "/home/xia/anaconda3/envs/xia-vgt/lib/python3.8/site-packages/detectron2/layers/init.py", line 3, in
from .deform_conv import DeformConv, ModulatedDeformConv
File "/home/xia/anaconda3/envs/xia-vgt/lib/python3.8/site-packages/detectron2/layers/deform_conv.py", line 11, in
from detectron2 import _C
ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory