Wraith
Wraith

Reputation: 11

Colab not recognizing local gpu

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

Answers (1)

bpinaya
bpinaya

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:

runtime

And then change it to GPU (takes some secs): GPU

Upvotes: 1

Related Questions