THCTensorMath.cu:26

在使用pytorch训练模型时遇到上面的问题，查了好久也不知道啥原因，最后在别人的帖子里找到类似的问题，就试试

解决办法：检查labels中是否有负值，如果有改成非负整数就可以了

相关资源

相似解决方案

热门标签

Java Python linux javascript Mysql C# Docker 算法前端 SpringBoot Redis Vue spring 设计模式 .net core .net kubernetes c++ 数据库数据结构大数据 js 机器学习微服务 Android Go 程序员面试 JVM ASP.net core 云原生人工智能后端 PHP git CSS golang k8s Nginx Django mybatis 深度学习多线程 React 架构 devops 爬虫云计算 Spring Boot LeetCode