You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Hi, I'm not the original poster but I am also having a similar problem. I am trying to train an image recognition AI but it will not use the GPU. the code snippet you provided evaluated to false, so I think that is the problem. However, I have tried to install cuda before, and it failed on the "Nsight Visual Studio Edition". I tried again and only selected the cuda files in the nvidia cuda installer. That showed it successfully installed, but it still won't work. I don't think it is very clear, so how exactly do I install cuda?
Hello,
I have all the gpu packages and no cpu ones yet It wont train via gpu usign the default code:
from imageai.Classification.Custom import ClassificationModelTrainer model_trainer = ClassificationModelTrainer() model_trainer.setModelTypeAsResNet50() model_trainer.setDataDirectory("idenprof") model_trainer.trainModel(num_experiments=100, batch_size=32)
The text was updated successfully, but these errors were encountered: