pytorch: CUDA out of memory
Liu-zz opened this issue · 1 comments
Liu-zz commented
您好!我在尝试运行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,不知道是什么原因,希望您能告诉我,谢谢!
qiuqiangkong commented
Hi,
The reason might be the GPU has been fully occupied by somebody else. You might try to switch another GPU.
Best wishes,
Qiuqiang
…________________________________
From: Liu-zz <notifications@github.com>
Sent: 08 April 2019 03:27
To: yongxuUSTC/sednn
Cc: Subscribed
Subject: [yongxuUSTC/sednn] pytorch: CUDA out of memory (#29)
您好!我在尝试运行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,不知道是什么原因,希望您能告诉我,谢谢!
―
You are receiving this because you are subscribed to this thread.
Reply to this email directly, view it on GitHub<https://eur02.safelinks.protection.outlook.com/?url=https%3A%2F%2Fgithub.com%2FyongxuUSTC%2Fsednn%2Fissues%2F29&data=02%7C01%7Cq.kong%40surrey.ac.uk%7C7cd18e3954fb4be6e6e308d6bbc9bc6b%7C6b902693107440aa9e21d89446a2ebb5%7C0%7C0%7C636902872457480031&sdata=XK4UiHwit3BQ80X3%2FNcE1NRPrdXvM11GtrNsdwjRjdU%3D&reserved=0>, or mute the thread<https://eur02.safelinks.protection.outlook.com/?url=https%3A%2F%2Fgithub.com%2Fnotifications%2Funsubscribe-auth%2FAMt5yWe-DUBpUWiw48YgefAu2VpvS44Jks5veqkKgaJpZM4chLL6&data=02%7C01%7Cq.kong%40surrey.ac.uk%7C7cd18e3954fb4be6e6e308d6bbc9bc6b%7C6b902693107440aa9e21d89446a2ebb5%7C0%7C0%7C636902872457490032&sdata=rXAtyMlJr9GfepagGOGxxmGO0yPI8qCJ8EmKt0hkWFs%3D&reserved=0>.