Pytorch-Correlation-extension
Pytorch-Correlation-extension copied to clipboard
Issues with CUDA 11.7
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
What should I do to make spatial-correlation-sampler run in such a setup? Thanks!