RuntimeError: CUDA out of memory(已解决)

今天用pytorch训练神经网络时,出现如下错误: RuntimeError: CUDA out of memory. Tried to allocate 144.00 MiB (GPU 0; 2.00 GiB total capacity; 1.29 GiB already allocated; 79.00 MiB free; 1.30 GiB reserved in total by PyTo
相关文章
相关标签/搜索