YOLOv6 O.D在colab运行时的AssertionError

问题遇到的现象和发生背景： 使用YOLOv6在定制数据集上训练，水下垃圾检测，在colab上运行，train时(finetune)遇到的问题是 assert torch.cuda.is_available() AssertionError，训练无法进行下去 （YOLOv6 可能是0.2.0版本)
运行结果及报错内容 ：train时遇到的问题是 assert torch.cuda.is_available() AssertionError，训练无法进行下去
我的解答思路和尝试过的方法 ：请敲打（修修补补）train.py文件或envs.py文件，后任何方式，使得训练能进行下去，--device 0 无助解决问题; nano,small,large模型的pt，都无法进行train,测试过了。
我想要达到的结果：请敲打（修修补补）train.py文件或envs.py文件，或任何方式，使得训练能进行下去

#我的codes，在colab上，不是本机cpu:

import os, sys
#下载数据集，underwater trash detection
!wget https://www.dropbox.com/s/lbji5ho8b1m3op1/reduced_label_yolov6.zip?dl=1
!unzip -q reduced_label_yolov6.zip?dl=1
!rm reduced_label_yolov6.zip?dl=1

if not os.path.exists('YOLOv6'):
!git clone https://github.com/meituan/YOLOv6.git

,,

,

img

#下载nano 最小的pt文件
!wget https://github.com/meituan/YOLOv6/releases/download/0.2.0/yolov6n.pt
..

img