解决方法：
安装与自己电脑配置匹配的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
相关推荐
《小目标目标检测的解决方法及方式》
《小目标目标检测的解决方法及方式》 最近在做小目标相关的项目，参考了一些博客、论文及书籍，在这里对小目标的方法和方式做了些总结。如果有哪些问题理解错误或补充欢迎讨论。 1.什么是小目标检测 在物体检测的各种实际
【原创】YOLOV4 训练自己的训练集——人头检测
数据集准备 首先下载人头检测数据集： 在 https://github.com/HCIILAB/SCUT-HEAD-Dataset-Release 中下载 SCUT_HEAD_Part_B 数据集，里面是图片&#x
mmdetectionV2.x版本 训练自己的VOC数据集
mmdetectionV2.x版本 训练自己的VOC数据集 1 首先根据规范的VOC数据集导入到项目目录下，如下图所示： mmdetection ----mmdet ----tools ----configs --
Nanodet训练自己的轻量级目标检测模型实践
官方地址:https://github.com/RangiLyu/nanodet 一、 准备训练的数据: 数据文件目录为如下,分别存放训练集和验证集的图片和xml文件(注意图片和标签一一对应) 二、建立自己yaml文件 在config文件
暂无评论