Google Colab怎么免费使用GPU实例

Google Colab默认使用的是CPU训练,Xeon双核的,性能较弱,也有免费的GPU和TPU实例可以选,怎么使用呢?

如果要更改运行时类型,依次点击菜单栏代码执行程序->更改运行时类型,硬件加速器改为GPU。TPU实例对免费用户来说几乎不可用。

改完之后重新运行代码即可。

查看所使用的GPU

!/opt/bin/nvidia-smi

Colab上用的最多的GPU型号应该是Tesla K80和Tesla T4。

查看使用的GPU序号

print(tf.config.experimental.list_physical_devices('GPU'))

Google Colab默认安装好了TensorFlow和PyTorch,用来做模型训练还是很方便的。

Leave a Comment

豫ICP备19001387号-1