BridgeDepthFlow
BridgeDepthFlow copied to clipboard
ImportError: libcudart.so.9.2
File "BridgeDepthFlow-master/models/networks/correlation_package/correlation.py", line 4, in
import correlation_cuda ImportError: libcudart.so.9.2: cannot open shared object file: No such file or directory
Hello, I tried this code with cuda-9.0 and cuda-10.0 with pytorch-1.1.0, and I have built and installed the correlation module. But when I run python train.py, it gets an error above, should I use cuda-9.2?
I've ever got this error before when using PyTorch-1.1.0. I suggest you to run with PyTorch-1.0.0 and Python-3.7.
@ltkong218 How do you solve this error? I met the same error even though I ran the code with PyTorch-1.0.0 and Python-3.7.
Use CUDA-9.2 and run
python setup.py build
python setup.py install
under ./models/networks/correlation_package and ./models/networks/resample2d_package respectively, then I can run the code.
I replace the correlation module with that in FlowNet2 https://github.com/NVIDIA/flownet2-pytorch
@ltkong218 Thanks for your reply.