LiUzHiAn / hf2vad

MIT License
121 stars 23 forks source link

about pre_process error #15

Open Smallwhite999 opened 2 years ago

Smallwhite999 commented 2 years ago

I strictly follow install.md is installed, but it always reports similar errors:

ImportError: libtorch_cuda_cu.so: cannot open shared object file, or: File "../pre_process/flownet_networks/correlation_package/correlation.py", line 4, in import correlation_cuda ModuleNotFoundError: No module named 'correlation_cuda'

I don't know what the reason is. Can you answer it, GPU: rtx3090, Ubuntu: 20, CUDA: 11.1, thank you very much!

LiUzHiAn commented 2 years ago

It seems the error is related to Flownet2, you may need to refer to their repo for the installation steps.