RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26 - 走看看

zoukankan html css js c++ java

RuntimeError: cuda runtime error (710) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26

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

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

查看全文

相关阅读:
jvm 更多链接
 JVM 内存初学 (堆(heap)、栈(stack)和方法区(method) )
python 排序
 python 第K个语法符号
 python conf.ini 文件的使用
 python 表示数字
 window 下安装redis
python redis相关操作
 python 矩阵乘法
 python 查找两个字符串a,b中的最长公共子串

原文地址：https://www.cnblogs.com/chuanyang/p/12803786.html

Copyright © 2011-2022 走看看