CLRNet icon indicating copy to clipboard operation
CLRNet copied to clipboard

Published 20 hours ago • verified publisher icon Turoad

OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

Open Miaoishappening opened this issue 2 years ago • 4 comments

Hi everyone,

I am trying to install some python packages with python setup.py build develop , but i do not have a GPU. so i installed pytorch with “conda install pytorch torchvision torchaudio cpuonly -c pytorch”.

Can i still try this CLRnet or a GPU is a necessary condition in your codes?

Miaoishappening avatar Sep 26 '22 13:09 Miaoishappening

NVIDIA GPU is necessary because NMS function is implemented in CUDA. You can overwrite the NMS function in Python, if you insist.

YatesZhang avatar Oct 19 '22 08:10 YatesZhang

@YatesZhang Thanks for replying. it helps a lot!

Miaoishappening avatar Oct 20 '22 09:10 Miaoishappening

@YatesZhang Thanks for replying. it helps a lot!

Hello！Could you please tell how to solve the problem？I met with the same problem，but I don't know how to overwrite the NMS function。Thanks a lot！

caolong-whu avatar Nov 04 '22 07:11 caolong-whu

NVIDIA GPU is necessary because NMS function is implemented in CUDA. You can overwrite the NMS function in Python, if you insist.

Hi would you mind tell me what should I do step by step I found these 2 source https://towardsdatascience.com/non-maxima-suppression-139f7e00f0b5 https://github.com/satheeshkatipomu/nms-python and after I overwrite nms what shoud I do?

anitach2003 avatar Jan 20 '23 01:01 anitach2003