CUDA out of memory

Question

CUDA out of memory

RajeshPandey057 opened this issue 4 years ago · comments

CUDA out of memory

Brandi Ortiz · Answer 1 · Tue Apr 06 2021 14:45:31 GMT+0800 (China Standard Time)

RuntimeError: CUDA out of memory. Tried to allocate 1.21 GiB (GPU 0; 3.95 GiB total capacity; 1.27 GiB already allocated; 889.19 MiB free; 1.27 GiB reserved in total by PyTorch)

Hongliang Zhang · Answer 2 · Fri Apr 23 2021 16:05:13 GMT+0800 (China Standard Time)

@oscarknagg @RajeshPandey057 @LIAOJUNJIE-OBJ , When I run the experiments.proto_nets on the dataset miniImageNet, it reports RuntimeError: CUDA out of memory. Though I have modified the parameters episodes_per_epoch and num_workers to make them smaller, this error still occurs. How can I resolve this problem? Thank you very much.