出现 CUDA out of memory 的解决方法

扎眼的阳光

2 年前

Table of Contents

1. 问题所示

运行这段程序的时候出现如下错误：

RuntimeError: CUDA out of memory. Tried to allocate 1.38 GiB (GPU 0; 7.80 
GiB total capacity; 5.94 GiB already allocated<

文章出处登录后可见！

已经登录？立即刷新