NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks
Other
3.08k stars 738 forks source link

Import Error: correlation_cuda #249

Open SwagJ opened 3 years ago

SwagJ commented 3 years ago

Dear All,

Current version: Pytorch: 1.4.0 CUDA: 10.1

Installation is successful without any warnings. However, when try to import correlation_cuda package, this error happens.

File "/disk_ssd/self-mono-sf/models/correlation_package/correlation.py", line 4, in import correlation_cuda ImportError: libcudart.so.10.0: cannot open shared object file: No such file or directory

I am not sure why correlation_cuda is trying to find libcudart.so.10.0 instead of libcudart.so.10.1

I appreciate all kinds of help. Thank you very much.

Best,