2 Colab之GPU

0 新项目

谷歌driver里面新建Colab文件，如下图所示：

如何设置GPU？
选择diam执行程序->更改运行时类型为GPU

1.验证GPU

查看是否为GPU设备

import tensorflow as tf

tf.test.gpu_device_name() # 查看设备名字

查看显卡情况

!/opt/bin/nvidia-smi # 查看显卡情况

输出：

至此为止，GPU设置完成！