解决运行出现CUDA error:out of memory的问题_outofmemoryerror: cuda out of memory.

作者：凡人多烦事01 | 2024-02-16 00:35:07

踩

outofmemoryerror: cuda out of memory.

模型运行出现以下问题，经过查找资料发现解决方案如下：
在这里插入图片描述
CUDA_VISIBLE_DEVICES限制一下使用的GPU。

比如有0,1,2,3号GPU，CUDA_VISIBLE_DEVICES=2,3，则当前进程的可见GPU只有物理上的2、3号GPU，此时它们的编号也对应变成了0、1，即cuda:0对应2号GPU，cuda:1对应3号GPU。

如何设置CUDA_VISIBLE_DEVICES:

① 使用python的os模块

import os

os.environ[‘CUDA_VISIBLE_DEVICES’]=‘2, 3’

②直接设置环境变量(linux系统)

export CUDA_VISIBLE_DEVICES=2,3

之后程序正常运行
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/88865