AssertionError: Torch not compiled with CUDA enabled

解决方法:
安装与自己电脑配置匹配的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

季南疯(JIM)

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

yolov4项目记录4-测试过程

目录 一、概述 二、测试过程 1.参数准备 2.定义模型 3.获取必要数据 4.输入模型 5.锚框筛选 ①利用物体置信度筛选 ②获取种类 ③按照物体置信度排序 ④非极大抑制 6.画出外接框 三、代码汇总 一、概述 训练