lannguyen0910 / food-recognition

🍔🍟🍗 Food analysis baseline with Theseus. Integrate object detection, image classification and multi-class semantic segmentation 🍞🍖🍕
MIT License
274 stars 76 forks source link

RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx #26

Closed alexhmyang closed 1 year ago

alexhmyang commented 1 year ago

RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

lannguyen0910 commented 1 year ago

Hi @alexhmyang , to load the weight you need to have GPU on your computer and more specifically, installed CUDA on it. If you don't have GPU then you should use my code on google colab. Sincere!

alexhmyang commented 1 year ago

thanks! where is colab code ? I havent fount it

lannguyen0910 commented 1 year ago

Here is the link, you can try it, but I saw that it has some issue with downloading the weights. So I will try to solve it asap!

jaykay9999 commented 1 year ago

i had a similar issue, chatgpt helped a bit but what worked for me is changing cuda:0 with cpu in the device parameter in the 3 .yaml files in in the config folder

lannguyen0910 commented 1 year ago

Yes, I forgot that I have included a device parameter in the config files. But I don't know if it works anymore in this version. Feel free to play with those configs and contact me if any issue happens. Thanks!