Issues with CUDA 11.7

Open huangzuo opened this issue 2 years ago • 1 comments

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!

Sep 11 '23 08:09 huangzuo

I resolved it by compiling it in CUDA 11.0 from the source codes. Pytorch version is 1.7.1.

Sep 14 '23 00:09 huangzuo