[STNNet] spatial-correlation-sampler lib issue with RTX 3000 series GPUs

[MounirB]

Hello,
It seems that the spatial-correlation-sampler doesn't work with a RTX 3000 series GPU when we try to install it through pip along with this conda environment
conda create -n STTNet python=3.6 pytorch=1.6 torchvision -c pytorch
I made it work by replacing python and pytorch by newest versions; respectively 3.9 and 1.11.

Do we need oldest versions of pytorch to make STNNet work ?

[tejasri19]

Could solve by running from source repo and python setup_cpu.py install

[sysu19351017]

I created an environment configured with CUDA-11.6, CUDNN-8.5.0, python-3.7.16, torch-1.12.0+cu116, and was able to run the codes successfully on the GTX3080. The Python libraries are as follows.

Besides, I can install the spatial-correlation-sampler lib just by pip command in the environment I created.