当前位置: 代码迷 >> 综合 >> RuntimeError: CUDA out of memory.
  详细解决方案

RuntimeError: CUDA out of memory.

热度:76   发布时间:2024-02-23 03:33:41.0

    Win10运行环境,python编程语言,运行一个pytorch框架的程序报错 “RuntimeError: CUDA out of memory. Tried to allocate 98.00 MiB (GPU 0; 8.00 GiB total capacity; 4.39 GiB already allocated; 5.27 MiB free; 89.58 MiB cached)”,解决办法:

    bachsize的参数值减小就解决了,举个例子:这里的主机运行内存是16GB,程序bachsize初始值设为16,减小为8还是报错,减小为8就成功解决了。

    

  相关解决方案