Cuda has memory error doing simplest calculation

1

Hello, I am a college student and pretty new to everything. I just got a Ubuntu laptop and set up CUDA for Pytorch, and it seems CUDA has no memory allocation.Showing me “CUDA error: Out of Memory”

Attached is the screenshot for my gpu.

Screenshot_20201227_201325
Screenshot_20201227_201325754×490 58.2 KB
Please help， thanks!!

2

(common cause) I think your model allocate the Max GPU memory, you’ll need to reduce batch size.
btw, could you please tell me what model did you train and batch size?

else follow this link
Frequently Asked Questions — PyTorch 2.1 documentation.