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,用来做模型训练还是很方便的。