作为机器学习初学者,又没钱买工作站,只能先用免费的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
GitHub Page:https://olig48.github.io/archives/a86fa85a.html