CUDA error: device-side assert triggered

[Millielele]

When I run train_net.py, I get this issue after loading R-50.pkl. I want to know how to solve that. Thanks a lot.
My environment: CUDA 11.1, python 3.7, torch 1.8.1

[LOVECCBACK]

请问你解决这个问题了嘛？我也是遇到这个问题并且环境版本跟你一样。