Cuda not available error when runnng setup.py

[henryzhao0615]

Hi,
I am trying to follow segmentation task. And I run sh ./make.sh on my GPU and it said that "Cuda is not available". However, I have cuda in my machine.
Here is the error:
(vrwkv)@vision3d2:~/research/Vision-RWKV/detection/ops$ sh ./make.sh
/home/users/u123/miniconda3/envs/vrwkv/lib/python3.10/site-packages/torch/cuda/init.py:128: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 11070). Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:108.)
return torch._C._cuda_getDeviceCount() > 0
No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'
Traceback (most recent call last):
File "/home/users/u123/research/Vision-RWKV/detection/ops/setup.py", line 70, in
ext_modules=get_extensions(),
File "/home/users/u123/research/Vision-RWKV/detection/ops/setup.py", line 43, in get_extensions
raise NotImplementedError('Cuda is not availabel')
NotImplementedError: Cuda is not availabel

But if I run "pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118" to install cuda. It said Requirement already satisfied.

I hope you can help me fix this issue. Thanks in advance!

[henryzhao0615]

Additionally, I found that for most of machines is not able to run "pip install torch==1.12.1+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
" according to the instructions, it will have error. And for mmcv-full, I tried "mim install mmcv-full==2.1.0" to solve. I tried these on two machines, one has a cuda version of 12.4 and one has 11.7. Both show the same result.

[BlinkDL]

i think it's fine to use latest pytorch & latest cuda

[henryzhao0615]

for the latest pytorch, mmcv-full 1.7 can not be installed.