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

[1codei]

trafficstars

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