目录

  • 1. 问题所示
  • 2. 原理分析
  • 3. 解决方法
    • 3.1 调batch_size大小
    • 3.2 不考虑梯度
    • 3.3 删除无用变量
    • 3.4 kill进程(暴力放弃)
    • 3.5 其他方法

1. 问题所示

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

RuntimeError: CUDA out of memory. Tried to allocate 1.38 GiB (GPU 0; 7.80
GiB total capacity; 5.94 GiB already allo
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。