Reputation: 11
Im trying to train a Neural Network that I wrote, but it seems that colab is not recognizing the gtx 1050 on my laptop. I can't use their cloud GPU's for this task, because I run into memory constraints
print(cuda.is_available())
is returning False
Upvotes: 1
Views: 1047
Reputation: 609
Indeed you gotta select the local runtime accelerator to use GPUs or TPUs, go to Runtime
then Change runtime type
like in the picture:
And then change it to GPU (takes some secs):
Upvotes: 1