原文:RuntimeError: CUDA error:out of memory的一种解决办法

具体环境: Untubu: . pytorch : . . CUDA: . 我是使用pytorch运行代码时,遇到了如下错误: RuntimeError: CUDA error:out of memory 我尝试看网上的很多方法,也没有解决方法,然后想到之前运行过一篇类似的代码,其中好像有这样的一行代码: 然后我抱着尝试的心态,在我的代码中加入下面俩行代码: 幸运的是,我的代码可以正常运行了。当然 ...

2021-11-08 11:25 0 849 推荐指数:

查看详情

解决RuntimeError: CUDA error: out of memory

解决RuntimeError: CUDA error: out of memory 跑代码时出现这个错误,输入以下命令 $ nvidia-smi -l 原来是卡0内存爆了,于是指定相对空闲的2卡 可以正常运行 ...

Fri Mar 12 05:12:00 CST 2021 0 595
存在空闲CUDA前提下报错:RuntimeError: CUDA error: out of memory

问题背景: 最近跑代码时发现报错CUDA out of memory,进入linux终端查看GPU使用情况(nvidia-smi),结果如下: 我用的GPU序号是0,但这块被人占用了,所以我可以用剩下的3号和4号。 解决方案: 在代码中更改GPU使用序号(修改/添加代码 ...

Fri Sep 17 17:36:00 CST 2021 0 111
解决CUDA out of memory

问题 用Pytorch跑模型时,会出现RuntimeError: CUDA out of memory 解决方法 1.真的没有内存了 可以改小batch_size 2.内存足够,不分配 Win10下多线程的问题,num_threads/workers=0解决 ...

Wed Apr 08 19:18:00 CST 2020 0 6006
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM