Bootstrap

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB解决办法

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 768.50 MiB already allocated; 16.39 MiB free; 9.50 MiB cached)解决方法

使用GPU出现上述错误,尝试改小batch_size的大小

主要是记录一下自己平时遇到的问题,和大家分享一下
如有侵犯,请联系我

点个赞支持一下吧

;