yongxuUSTC/sednn

pytorch: CUDA out of memory

Liu-zz opened this issue · 1 comments

您好!我在尝试运行pytorch部分代码时会提示“CUDA out of memory”,具体如下:
File "/sednn/mixture2clean_dnn_pytorch/prepare_data.py", line 335, in log return torch.log(x + eps) RuntimeError: CUDA out of memory. Tried to allocate 3.50 MiB (GPU 0; 15.90 GiB total capacity; 15.02 GiB already allocated; 1.88 MiB free; 62.70 MiB cached) ('cuda:', True)
在CPU上运行时正常
所用数据仅仅为mini_data,不知道是什么原因,希望您能告诉我,谢谢!