RuntimeError: CUDA error: device-side assert triggered CUDA kernel errors might be asynchronously re

作者：知新_RL | 2024-02-16 01:08:55

踩

runtimeerror: cuda error: device-side assert triggered cuda kernel errors mi

要么是device那里没有指定正确的CUDA
要么是网络维度没有对齐
检查一下torch中device语句
然后检查一下，我的错误就是维度没有对齐，定义模型的时候将维度写死了，然后在train的时候输入维度不是网络那个维度了，比我我在模型的输入时候多写了个permute导致维度改变
如果实在不清楚维度的话就从输入开始一步步打印shape即可找到问题

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/知新_RL/article/detail/89198