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)

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

暂无评论

发表评论

相关推荐

【学习分享】目标检测中的锚框(Anchor)

锚框
引例
在理解目标检测的锚框之前,我们首先通过一个不太严谨的例子对锚框进行一个简单的了解: 由于目前污染比较严重,导致海洋中漂浮着许多垃圾,这些垃圾既污染环境,又不利

SSD网络及代码理解

写在前面:在读了SSD的论文之后,完整的看了一遍SSD的代码,有了许多体会,以此记录自己的学习过程。 论文传送门:SSD: Single Shot MultiBox Dete