d-li14/involution

CUDA version error: the provided PTX was compiled with an unsupported toolchain.

hheavenknowss opened this issue · 0 comments

Thanks your work, it's very useful, and I've used involution Pytorch version for a while, but recently I need to try a bigger feature than before, and my GPU memory is not enough, so I try to use CUDA version, but here is an error I couldn't find a solution on the internet:
File "cupy/cuda/function.pyx", line 241, in cupy.cuda.function.Module.load
File "cupy/cuda/function.pyx", line 243, in cupy.cuda.function.Module.load
File "cupy_backends/cuda/api/driver.pyx", line 246, in cupy_backends.cuda.api.driver.moduleLoadData
File "cupy_backends/cuda/api/driver.pyx", line 124, in cupy_backends.cuda.api.driver.check_status
cupy_backends.cuda.api.driver.CUDADriverError: CUDA_ERROR_UNSUPPORTED_PTX_VERSION: the provided PTX was compiled with an unsupported toolchain.

it seems is a CUDA version error, but I really don't know how to solve it, could you please help me about this?
Here is my CUDA info:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

NVIDIA-SMI 455.23.04 Driver Version: 455.23.04 CUDA Version: 11.1

Thanks a lot!