RuntimeError: CUDA out of memory. Tried to allocate xx MiB

RuntimeError: CUDA out of memory. Tried to allocate 86.00 MiB
几种可能原因:

  1. batch size 过大,设置小一些就可以了(另外,batch size的设置应该是GPU使用数量的整数倍,比如4块GPU应设置为4,8,12,16,24等)
  2. GPU使用数量问题,有的程序默认1个,且不能修改,改了就报这个错
  3. 内存占用过高,清理内存,杀死Python进程

你可能感兴趣的:(笔记)