RuntimeError: CUDA error: out of memory的解决【实测成功】

最新推荐文章于 2024-06-13 23:10:33 发布

captain飞虎大队

最新推荐文章于 2024-06-13 23:10:33 发布

阅读量4.8w

点赞数 44

分类专栏： python

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.csdn.net/weixin_41529093/article/details/115144349

版权

python 专栏收录该内容

453 篇文章 18 订阅

订阅专栏

仅作为记录，大佬请跳过。

文章目录

背景
解决
参考
原因

背景

博主使用linux服务器运行MIL_train.py程序时，出现RuntimeError: CUDA error: out of memory的错误（之前运行这个python木有问题）

解决

在MIL_train.py文件里加入：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

即可。

参考

感谢大佬博主文章：传送门

在这里插入图片描述

原因

服务器的默认gpu显卡正在使用，需要使用gpu的另一个空闲的显卡；linux里查看gpu的显卡使用状态用代码：

nvidia-smi

在这里插入图片描述

运行成功后：

在这里插入图片描述

captain飞虎大队

关注

44
点赞
踩
100

收藏

觉得还不错? 一键收藏
13
评论
RuntimeError: CUDA error: out of memory的解决【实测成功】

仅作为记录，大佬请跳过。文章目录背景解决参考原因背景博主使用linux服务器运行MIL_train.py程序时，出现RuntimeError: CUDA error: out of memory的错误（之前运行这个python木有问题）解决在MIL_train.py文件里加入：import osos.environ["CUDA_VISIBLE_DEVICES"] = '1'即可。参考感谢大佬博主文章：传送门原因服务器的默认gpu显卡正在使用，需要使用gpu的另一个空闲的显卡；li
复制链接

扫一扫

专栏目录

评论 13

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。