yolov3-channel-and-layer-pruning 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)

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)

Open huangzicheng opened this issue 4 years ago • 4 comments

gpu have enough space,but it always show this matter

Mar 29 '20 03:03 huangzicheng

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

Mar 29 '20 03:03 LeonNerd

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

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

Mar 29 '20 04:03 huangzicheng

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

Jun 01 '20 07:06 qaazii

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

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

大哥能否仔细说哈。。。

May 15 '21 07:05 Building-max