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
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