pytorch -- RuntimeError: CUDA out of memory. Tried to allocate 1.88 GiB

提示我的GPU free在1.05 GiB

修改batch_size大小，之前是24，现在修改成15就可了

posted @ 2020-08-09 18:21  aha是Q啊  阅读(705)  评论(0)    收藏  举报