2022年12月27日
yolo5使用gpu时遇到的问题记录
摘要： 一、问题描述： 1、训练的时候提示不支持gpu 2、使用如下命令检查为False import torch torch.cuda.is_available() 二、原因： pytorch版本的问题 三、解决办法： 重新安装符合的版本，参考的是： torch.cuda.is_available() 解 阅读全文
posted @ 2022-12-27 11:47 星星c# 阅读(1051) 评论(0) 推荐(0) 编辑