search

lelimite4444 / BridgeDepthFlow

Bridging Stereo Matching and Optical Flow via Spatiotemporal Correspondence, CVPR 2019
120 stars 20 forks source link

ImportError: libcudart.so.9.2 #2

Open ltkong218 opened 5 years ago

ltkong218 commented 5 years ago

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?

lelimite4444 commented 5 years ago

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.

wanghao14 commented 4 years ago

@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.

wanghao14 commented 4 years ago

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.

ltkong218 commented 4 years ago

I replace the correlation module with that in FlowNet2 https://github.com/NVIDIA/flownet2-pytorch

wanghao14 commented 4 years ago

@ltkong218 Thanks for your reply.