Colab使用教程

目录

• 有关链接
• 使用GPU
• 切换文件夹
• 参考

有关链接

• Google Colabratory
• Google Drive

使用GPU

以下两种方式都可以：

1. "修改"->"笔记本设置"->"硬件加速器"选择GPU
2. "代码执行程序"->"更改运行时类型"->"硬件加速器"选择GPU

然后运行以下代码确认GPU是否正常运行

import tensorflow as tf
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
  raise SystemError('GPU device not found')
else:
  print(device_name)

运行正常可以看到如下结果：

否则看到如下结果：

或者使用命令

!nvidia-smi

可以看到以下结果：

否则看到如下结果：

PS：如果你有其他会话正在运行，可能会看到这样的提示

切换文件夹

不能用Linux下常用的cd命令，要用如下命令

import os
os.chdir("/content/drive/My Drive/your_path")

参考

转载几篇自己使用的时候参考的博客，方便以后查阅。

• Colab 实用教程

• 使用colab运行深度学习gpu（Mask R-CNN）实践

• Google Colab——用谷歌免费GPU跑你的深度学习代码