Colab 全称Colaboratory,是谷歌提供的一个在线工作平台,可以与谷歌云盘协作使用。我们可以在Colab平台上运行代码,而且大部分常用的包都已经安装好,不需要再进行安装,也不需要进行环境配置,非常方便快捷,对于初学者来说非常友好。
最重要的一点是,在Colab上可以免费使用GPU,具体的使用方法我会在后面的内容中详细说明。我想可以使用免费的GPU是大多数同学选择Colab的主要原因,对于日常的代码学习来说是足够用的,如果不够用还可以购买Colab Pro,可以获得更多稳定的计算单元。让我们开始吧!
1.前期准备
首先,想要使用谷歌的服务就必须要有一个谷歌账号,谷歌账号有免费的15G云盘空间,并且可以和Colab协作使用,非常方便。其次,由于谷歌账号的注册需要手机号,而谷歌并不支持中国地区的手机号,因此需要虚拟手机号注册一个谷歌账号进行使用。当然,在整个申请账号和使用Colab的过程中,都需要科学上网。下面我们来看看如何申请谷歌账号。
开启科学上网后,打开以下网址,购买一定量的额度后,选择一个地区的手机号。需要注意的是,申请的虚拟手机号有效期是20分钟,需要在20分钟内注册号谷歌账号。具体的申请过程见网页。
当我们成功登录谷歌的Colab之后,就会进入以下界面:
可以在设置下面设置字体、主题样式等:
既然已经成功进入Colab了,就让我们来看看如何使用它吧。
2.COlab的使用(如何免费使用GPU、GAN模型训练等)
接下来我将演示如何使用本地手写数据集对生成对抗神经网络进行训练,生成手写数字。代码和数据集在以下链接:
链接:百度网盘 请输入提取码
提取码:eilc
首先我们需要使用之前申请的谷歌账号登录谷歌云盘,显示我们有免费的15G云存储空间:
然后打开云盘,将已经编写好的神经网络代码和训练集放在云盘的同一个目录下:
右键需要运行的代码,选择使用Colab打开,就会自动跳转到Colab:
打开时候需要在主程序前运行以下两个代码,这两个代码的作用分别是加载谷歌云盘中的数据并连接到Colab,同时将转到工作文件夹下:
from google.colab import drive drive.mount('/content/drive/')
%cd /content/drive/My Drive/Colab Notebooks/ !ls
弹出账号登陆,单击自己的谷歌账号,并点击允许:
点击两个图标更新文件夹:
将代码中的训练集地址改成前面运行的代码块的结果:
设置GPU进行训练,点击视图下的笔记本信息——>打开笔记本设置——>修改为使用GPU——>点击保存:
点击运行代码,模型开始训练:
最终训练结果: