Weird import error #117
Description
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