RuntimeError: CUDA error: an illegal memory access was encountered

xingyizhou / CenterNet

Object detection, 3D detection, and pose estimation using center point detection:

MIT License

7.29k stars 1.93k forks source link

RuntimeError: CUDA error: an illegal memory access was encountered #1044

Open AlGo1004 opened 4 months ago

AlGo1004 commented 4 months ago

When I python src/demo.py ctdet --demo images/16004479832_a748d55f21_k.jpg --load_model models/ctdet_coco_dla_2x.pth, disappeared this error, how can I solve it? My environment is as follows： pytorch=1.7.0 cuda=11.0.0