1
模型应用
在使用DCU推理yolov5模型时遇到问题
2024-12-16 10:53:21
166

操作系统：ubuntu20.04
DTK版本：24.04.3
驱动版本：rock-5.7.1-6.2.26
DCU类型：z100l
异常描述：在使用yolov5推理代码detect推理时，遇到报错“AssertionError: CUDA unavailable, invalid device 0 requested”x需要帮助：需要dcu显卡的调用方法或者适应dcu版本的yolov5推理模型

全部评论（3）
只看楼主
L
liumg
2楼

python3

>>import torch

>>print(torch.cuda.is_available())

环境检查正常么
2024/12/17
L
liumg
1楼
帮忙把torch的版本信息发一下
2024/12/16
网
网友pJIem5
回复 @liumg
2024/12/16
没有更多内容了
--
楼主
Level.1
0 帖子
楼主热帖
暂无数据
常见问题新帖
暂无数据