Colab无法识别本地GPU

Question

我正在尝试训练我编写的神经网络，但是colab似乎无法识别笔记本电脑上的gtx 1050。我无法使用他们的云GPU来执行此任务，因为遇到内存限制

print(cuda.is_available())

返回False

Answer 1

实际上，您必须选择本地运行时加速器以使用GPU或TPU，依次转到Runtime和Change runtime type，如图所示：

然后将其更改为GPU（需要几秒钟）：