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

fcjian / InstaGen

InstaGen: Enhancing Object Detection by Training on Synthetic Dataset, CVPR2024

https://fcjian.github.io/InstaGen

MIT License

68 stars 2 forks source link

Open YasminZhang opened 3 weeks ago

YasminZhang commented 3 weeks ago

When I was running sh instagen_scripts/infer_detector.sh, I encountered an error

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

Can you tell me your pytorch version and cuda version? I searched online and found this is a cuda pytorch version issue.