具体错误如下：
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

解决方法：
1.根据提示，打开对应文件夹下的py文件
在这里插入图片描述
2.在此处添加".cpu()"即可
在这里插入图片描述

# pycharm # pytorch # ide
Logo
https://edu.csdn.net/learn/39067/627173?utm_source=2019755004

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

程序员鼓励师～～Cursor AI Rules - 宪法驱动的超级AI编程伙伴

每种都有独特的交互风格、语言模式和行为特征，让你可以根据心情、任务类型和个人偏好选择最适合的AI伙伴。- 详细角色描述和使用示例。，让呼叫变得更加亲切自然！

avatar AI编程社区
cover

【100% AI编程二】KitfoxMarkdown：Markdown 文件（上云）链接转换工具

avatar AI编程社区
cover

从零开始安装并配置开源AI编程神器OpenCode

avatar AI编程社区