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

[LeonNerd]

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

[huangzicheng]

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

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

[qaazii]

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]

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

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

大哥能否仔细说哈。。。