yongxuUSTC / sednn

deep learning based speech enhancement using keras or pytorch, make it easy to use
http://staff.ustc.edu.cn/~jundu/The%20team/yongxu/demo/SE_DNN_taslp.html
334 stars 124 forks source link

pytorch: CUDA out of memory #29

Open Liu-zz opened 5 years ago

Liu-zz commented 5 years ago

您好!我在尝试运行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 5 years ago

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 GitHubhttps://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 threadhttps://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.