Pytorch-Correlation-extension icon indicating copy to clipboard operation
Pytorch-Correlation-extension copied to clipboard

Issues with CUDA 11.7

Open huangzuo opened this issue 2 years ago • 1 comments
trafficstars

I tried to implement v4.0.0 in the following environment: CUDA 11.7 on A100 Gpu Torch 1.12.1 in Linux. Installation using pip is successful but running in python failed. Errors: from .spatial_correlation_sampler import SpatialCorrelationSampler, spatial_correlation_sample File "/home/xxx/venv/lib/python3.10/site-packages/spatial_correlation_sampler/spatial_correlation_sampler.py", line 6, in import spatial_correlation_sampler_backend as correlation ImportError: libc10_cuda.so: cannot open shared object file: No such file or directory.

What should I do to make spatial-correlation-sampler run in such a setup? Thanks!

huangzuo avatar Sep 11 '23 08:09 huangzuo