RuntimeError: CUDA out of memory.

    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还是报错,减小为4就成功解决了。

    

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页