torchvision报错

报错：NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend. This could be because the operator doesn't exist for this backend
torch、torchvision、torchaudio都有GPU版本，有时报错可能是装的CPU版本，最好按官网命令统一装，重新装torchvision即解决

posted @   shanqiu24  阅读(2846)  评论(0)  编辑  收藏  举报
相关博文：
· 土堆Pytorch教程笔记
· 解决notebook中显示图片代码在pycharm不起作用的问题
· Pytorch错误：Torch not compiled with CUDA enabled
· 验证torch和torchvision安装成功
· torch找不到CUDA的问题
阅读排行：
· Manus重磅发布：全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后，我竟然真的恢复了删除的微信聊天记录！
· 没有Manus邀请码？试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
点击右上角即可分享
微信分享提示