使用colab跑机器学习

作为机器学习初学者,又没钱买工作站,只能先用免费的colab了。colab是谷歌提供的免费服务器有16G内存,还有免费的GPU用,基本完爆我的轻薄本。效果如下图,但是使用colab还是需要一些简单的配置,方便训练模型,保存数据。

使用colab跑机器学习

挂载谷歌硬盘

colab自带100G的空间,但是重连后,会清理空间,把数据保存在谷歌云盘里更为方便。之间都是用命令操作,有点麻烦,后来发现能直接点击文件夹装载,谷歌云盘就被挂载在/content/drive/MyDrive目录了,如下图。

使用colab跑机器学习

添加环境变量

在colab有时需要使用自己封装的代码,这时就需要添加环境变量了,这里使用sys,代码如下

import sys
sys.path.append("/content/drive//") #自定义环境变量的位置

打开GPU

colab默认不开启GPU,需要点击修改笔记本设置

使用colab跑机器学习

然后选择GPU或者TPU

使用colab跑机器学习

需要说明的是,GPU一开始能分配到好的,用的多了,就分配不到了,建议多创几个google账号,就能用更多GPU资源了。

切换tensorflow版本

colab默认使用tensorflow2.x版本,需要使用1.x的时候可以用内置命令更换

#先切换到GPU或TPU,然后使用以下命令
%tensorflow_version 1.x

GitHub Page:https://olig48.github.io/archives/a86fa85a.html

上一篇:colab使用


下一篇:迁移学习 colab 完整示例:fruits-360 数据集