search

WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
23.25k stars 8.01k forks source link

pytorch_object_detection /faster_rcnn的代码运行问题 #821

Closed Zhangjq7585 closed 4 months ago

Zhangjq7585 commented 5 months ago

1 2 博主您好，我在运行你的faster_rcnn代码的时候，运行train_res50_fpn.py报错，RuntimeError: CUDA out of memory. Tried to allocate 476.00 MiB (GPU 0; 9.77 GiB total capacity; 7.69 GiB already allocated; 242.69 MiB free; 7.92 GiB reserved in total by PyTorch)。我是租AutoDL服务器，显存8G，这个地方是要租一个显存更大的卡吗？ 我也看过您的视频，batchsize可以调小，但是我不太理解这里想和你探讨一下。