search

TexasInstruments / edgeai-modelmaker

This repository has been moved. The new location is in https://github.com/TexasInstruments/edgeai-tensorlab
https://github.com/TexasInstruments/edgeai
Other
1 stars 0 forks source link

ERROR: NO CUDA runtime is found, using CUDA_HOME=‘****’ but CUDA is Available! #10

Closed YGZone closed 11 months ago

YGZone commented 1 year ago

When I try to train example in modelmaker, the project only use CPU to train, but my CUDA is available! image

bash info is this: No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda-11.3/'

When I use python to run code: import torch torch.cuda.is_available() the result is True, and then I try to run a deeplearning project to test my gpu, image

result is that the gpu could work well, I don't konw where is the error. Could you give me some ideas, thanks!

YGZone commented 1 year ago

I solve this problem by edit config file , when I set gpu = 1 ,the cuda could work.