Google Colaboratory的配置和使用
0. 前言
最近一直在炼丹,之前老师租的服务器上千块钱一个月,显卡也很一般,Google完全免费的Colaboratory,通常给到的GPU资源是Tesla T4,TPU也是能免费用的,8个核,相当于8个GPU分布式训练,快的飞起,就是用起来程序上很麻烦。开通Colab Pro每月就9.9刀,同时可以多开,拿到的GPU一般的Tesla V100和P100,相比T4快了不少,V系类算力更高些。
谷歌在环境方便做了很好的适配,兼容性很高,用了一段时间之后,只能说Google一次又一次让我感到震惊,友好的UI和全局中文,添加了很多方便的可视化操作,从使用体验上来说就已经无敌了,这就是资本的力量吗。
当然Colab有使用限制,每次运行最多12个小时,不过我Pro三个会话好像通常不到6个小时,做些小模型小数据集的实验还是比自己的服务器划算一些.
1. 创建笔记本
新建Google Colaboratory
如果Colaboratory不存在,转到页面https://colab.research.google.com/notebooks/welcome.ipynb
点击左上角的“复制到云盘”,重新进入自己的云盘新建选项中就会出现Colaboratory。
然后可以新建ipynb文件,也就和jupyter notebook类似。
在ipynb中可以对当控制台使用,在指令面前加"!"就行。
2. 选择GPU/TPU
在“修改”=>“笔记本设置” 中可以选择GPU,然后在右边有连接选项
检测GPU情况:
1 | # 检测GPU情况 |
查看显卡信息:
1 | !nvidia-smi |

3. 挂载Google Drive
在左边的控制栏中有直接的挂载Drive的选项,也可以通过drive.mount
添加。
1 | from google.colab import drive |
4. 更改运行目录
更改到目录之后可以直接!python youcode.py
运行python程序。
1 | import os |
5. 切换tensorflow环境
1 | %tensorflow_version 1.x |