RuntimeError: CUDA out of memory. Tried to allocate 22.00 MiB (GPU 0; 10.91 GiB total capacity; 10.03 GiB already allocated; 18.12 MiB free; 10.17 GiB reserved in total by PyTorch)

tanluren / yolov3-channel-and-layer-pruning

yolov3 yolov4 channel and layer pruning, Knowledge Distillation 层剪枝，通道剪枝，知识蒸馏

Apache License 2.0

1.5k stars 446 forks source link

Open huangzicheng opened 4 years ago

huangzicheng commented 4 years ago

gpu have enough space,but it always show this matter

LeonNerd commented 4 years ago

剩余空间不足调低batchsize 或者下载apex

huangzicheng commented 4 years ago

原因找到啦：不能直接pip install apex 卸载从源码安装

pip uninstall apex git clone https://www.github.com/nvidia/apex cd apex python setup.py install

qaazii commented 4 years ago

How you solved Out of memory problem. I am still struggling with this even though I have GPU 1080Ti and I installed apex

Building-max commented 3 years ago

原因找到啦：不能直接pip install apex 卸载从源码安装

pip uninstall apex git clone https://www.github.com/nvidia/apex cd apex python setup.py install

大哥能否仔细说哈。。。