No CUDA runtime is found

[mirzaqasem]

hi
first i thank you for your code
but when i run

git clone https://github.com/megvii-research/TreeEnergyLoss
cd TreeEnergyLoss
pip install -r requirements.txt
cd kernels/lib_tree_filter
sudo python3 setup.py build develop

i got error

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'
Traceback (most recent call last):
File "/content/FedICRA/code/utils/TreeEnergyLoss/kernels/lib_tree_filter/setup.py", line 21, in
raise Exception('This implementation is only avaliable for CUDA devices.')
Exception: This implementation is only avaliable for CUDA devices.

but i active GPU on runtime and torch.cuda.is_available() return true

please help me to fix it