加载中...

使用colab跑机器学习


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

挂载谷歌硬盘

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

添加环境变量

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

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

打开 GPU

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

然后选择 GPU 或者 TPU

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

切换 tensorflow 版本

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

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

文章作者: Muryor
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 Muryor !
评论
  目录