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

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

你可能感兴趣的:(坑)