问题遇到的现象和发生背景: 使用YOLOv6在定制数据集上训练,水下垃圾检测,在colab上运行,train时(finetune)遇到的问题是 assert torch.cuda.is_available() AssertionError,训练无法进行下去 (YOLOv6 可能是0.2.0版本)
练习使用YOLOv6 在定制数据集(水下垃圾识别),见https://learnopencv.com/yolov6-custom-dataset-training/?ck_subscriber_id=1715937919
运行结果及报错内容 :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
,,
,
#下载nano 最小的pt文件
!wget https://github.com/meituan/YOLOv6/releases/download/0.2.0/yolov6n.pt
..