本期博客,手把手带你使用谷歌的Colab云平台服务器,进行深度学习训练。
使用前提:(科学上网)
1.登录
打开此链接,然后点击右上角的文件,选择新建笔记本,进入到colab界面,如下图所示;
2.使用colab的免费GPU
进入上图的界面后,要使用colab的免费GPU,点击上图中的修改,然后选择笔记本设置,硬件加速器设置成GPU,保存即可。
并且此处可以执行以下代码,检测一下分配到的GPU类型,一般免费的GPU会分配Tesla-T4的GPU供你使用,速度只能说很一般,建议升级成colab pro用户;pro用户现在可以使用V100或者A100的GPU。
!nvidia-smi
3.关联云盘
在GPU设置完成后,选择关联云盘,这一步十分重要!!!建议一定要关联云盘并将项目文件上传至云盘中,然后直接在colab中打开使用;运行下面的代码进行云盘关联:
from google.colab import drive
drive.mount('/content/drive/')
运行完毕后,会自动弹出是否关联的对话框,点击连接到Google云端硬盘即可,接下来还会有一个确认连接的弹框自动出现,点击即可
接下来可以像linux一样去使用了,注意在运行代码时,前面要加英文的!,例如:
!python train.py #前面必须加英文感叹号
4.注意的细节
4.1进入文件路径
在关联完毕后,要执行云盘中的某一个项目,要先进入该项目的路径下,并安装所需库,才能开始进行操作,例如:
复制项目的路径,然后在前面加上%cd,我的项目文件叫yolov5-6.1-ds,大家可以右键项目复制路径后,%cd该路径,进入项目中,然后再执行一系列操作,YOLOV5中还需要执行以下代码,来安装其所需要的库或者安装包,然后就可以尽情使用了!!!
!pip install -r requirements.txt
文章出处登录后可见!
已经登录?立即刷新