刚开始遇到这个错误还以为硬件哪里出问题了,后面发现是将二分类模型的out_dim从2修改到1导致的,不过在修改out_dim之后仍然报错,这可能是因为内存泄漏或其他一些底层bug导致,在restart kernel之后,报错消失
RuntimeError: CUDA error: device-side assert triggered Compile with `TORCH_USE_CUDA_DSA` to enable d
最新推荐文章于 2024-05-13 10:42:06 发布
刚开始遇到这个错误还以为硬件哪里出问题了,后面发现是将二分类模型的out_dim从2修改到1导致的,不过在修改out_dim之后仍然报错,这可能是因为内存泄漏或其他一些底层bug导致,在restart kernel之后,报错消失