Weird import error

[svyslf]

trafficstars

Hi, i'm using this package as a dependency for the STNNet density map estimation model. It was working fine yesterday, but today it's failing. I'm working in google colab, and tried reinstalling with the following commands: !pip install spatial-correlation-sampler --no-cache-dir !pip install spatial-correlation-sampler --no-build-isolation

My torch install details: Name: torch Version: 2.6.0+cu124 Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration Home-page: https://pytorch.org/ Author: PyTorch Team Author-email: [email protected] License: BSD-3-Clause Location: /usr/local/lib/python3.11/dist-packages

i don't even understand what the issue here is or what is causing it - would appreciate help!

---

ImportError Traceback (most recent call last) in <cell line: 0>() 17 # from grad_cam import CAM 18 import torchvision.transforms.functional as tff ---> 19 from stnmodel import STNNet 20 from loss_func import calc_loc_loss, calc_trk_loss 21 from utils import save_checkpoint

2 frames /content/gdrive/MyDrive/DroneCrowd-implementation/STNNet/stnmodel.py in 2 import torch 3 from torchvision import models ----> 4 from spatial_correlation_sampler import SpatialCorrelationSampler 5 from pointconv import PointConvDensitySetAbstraction 6 import numpy as np

/usr/local/lib/python3.11/dist-packages/spatial_correlation_sampler/init.py in ----> 1 from .spatial_correlation_sampler import SpatialCorrelationSampler, spatial_correlation_sample

/usr/local/lib/python3.11/dist-packages/spatial_correlation_sampler/spatial_correlation_sampler.py in 4 from torch.nn.modules.utils import _pair 5 ----> 6 import spatial_correlation_sampler_backend as correlation 7 8

ImportError: /usr/local/lib/python3.11/dist-packages/spatial_correlation_sampler_backend.cpython-311-x86_64-linux-gnu.so: undefined symbol: _ZN3c106detail23torchInternalAssertFailEPKcS2_jS2_RKNSt7__cxx1112basic_stringIcSt11char_traitsIcESaIcEEE