CUDA version with CUDA 11.4 and CUDA 9.0

[YerongLi2]

Yes, I also got the CUDA version issue, the error message I am getting is

ImportError: /home/wentaoy2/LTVRR/lib/model/roi_pooling/_ext/roi_pooling/_roi_pooling.so: undefined symbol: __cudaRegisterFatBinaryEnd

Which CUDA version are you using?
And the version of CUDA and pytorch I am on is

$ pip list| grep torch
torch 0.4.1.post2
torchvision 0.1.8
$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Mon_Oct_11_21:27:02_PDT_2021
Cuda compilation tools, release 11.4, V11.4.152
Build cuda_11.4.r11.4/compiler.30521435_0
(vr) wentaoy2@bock:~/LTVRR$

I also tried CUDA-9.0 and got simiar issures:

File "/home/wentaoy2/LTVRR/lib/model/roi_pooling/functions/roi_pool.py", line 3, in <module>
   from .._ext import roi_pooling
File "/home/wentaoy2/LTVRR/lib/model/roi_pooling/_ext/roi_pooling/__init__.py", line 3, in <module>
   from ._roi_pooling import lib as _lib, ffi as _ffi
ImportError: /home/wentaoy2/LTVRR/lib/model/roi_pooling/_ext/roi_pooling/_roi_pooling.so: undefined symbol: __cudaRegisterFatBinaryEnd

    $ nvcc --version
    nvcc: NVIDIA (R) Cuda compiler driver
    Copyright (c) 2005-2017 NVIDIA Corporation
    Built on Fri_Sep__1_21:08:03_CDT_2017
    Cuda compilation tools, release 9.0, V9.0.176