解决方法：
安装与自己电脑配置匹配的pytorch
检测cuda是否可用：
import torch
print(torch.cuda.is_available())
运行结果：
True
版权声明：本文为CSDN博主「季南疯(JIM)」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_46235937/article/details/121885819
相关推荐
目标检测：从头开始搭建YOLOv3 SPP 原理
一、原理解析
论文地址： https://arxiv.org/abs/1506.02640 (YOLO v1) https://arxiv.org/abs/1612.08242 (YOLO v2) https://arxiv.o
【学习分享】目标检测中的锚框(Anchor)
锚框
引例
在理解目标检测的锚框之前，我们首先通过一个不太严谨的例子对锚框进行一个简单的了解： 由于目前污染比较严重，导致海洋中漂浮着许多垃圾，这些垃圾既污染环境，又不利
目标检测之牛仔行头检测（上）—— 读取coco数据集并转换为yolo数据格式，以YOLOV5为baseline训练并提交结果
目标检测之牛仔行头检测（上）—— 读取coco数据集，可视化处理，转换为yolo数据格式 目录
目标检测之牛仔行头检测（上）—— 读取coco数据集ÿ
raise RuntimeError(‘Attempting to deserialize object on a CUDA ‘ RuntimeError: Attempting to deseri
raise RuntimeError(Attempting to deserialize object on a CUDA
RuntimeError: Attempting to deserialize object on a CUDA de
暂无评论