colab使用方法(利用免费的GPU进行模型的训练)

一、首先开启魔法

这里就不展示了。

二、打开网站

https://colab.google/

在这里插入图片描述

点击open colab。
在这里插入图片描述
点击修改中的笔记本设置,选择GPU加速。

因为colab无法上传文件夹,首先需要将自己的代码压缩一下

在这里插入图片描述

1、点击文件 2、点击上传刚刚压缩好的代码。

三、模型的训练

在这里插入图片描述

首先,点击左上角的+代码,添加一行命令用于将压缩的文件解压。-d参数用于确定解压后的代码的位置。上图所示的文件所在位置默认为/content/。

!unzip /content/yolov5-6.0.zip -d /content/yolov5

再切换到当前目录

%cd /content/yolov5/yolov5-6.0

根据代码中的requirements.txt下载需要的依赖库

!pip install -r requirements.txt

可以使用tensorboard进行可视化(可选)

%load_ext tensorboard
%tensorboard --logdir=runs/train

最后,运行代码

!python train.py
  • 9
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值