热门标签
热门文章
- 1计算机网络实验二（常用网络管理命令1）_计算机网络实验windows系统网络服务器管理
- 2穷举法解锁华为手机bootloader_dcunlocker账号
- 3【毕业设计】基于JAVA的springboot地方废物回收机构管理系统(源代码+论文）_垃圾废品回收系统源码
- 4tuxerantfs2022Mac上读写NTFS格式硬盘，快捷的访问、编辑、存储和传输文件_tuxera2022 ntfs for mac怎么使用
- 5一年创作之旅：点滴积累，感悟成长
- 6HarmonyOs源码获取编译烧录_harmonyos 源码编译
- 7[2018年工作重点规划] substance designer 使用和工作流，材质库
- 8论文查重遇上AI辅写疑似度，七招助你化解危机_论文降低aigc
- 9nginx 二级目录部署vue项目_nginx 二级目录部署项目
- 10警告解决方案：To copy construct from a tensor, it is recommended to use..._data.tensordatasetto copy construct from a tensor,
当前位置: article > 正文
RuntimeError: CUDA out of memory解决方法_runtimeerror: cuda out of memory.
作者：羊村懒王 | 2024-03-18 14:05:10
赞
踩
runtimeerror: cuda out of memory.
RuntimeError: CUDA out of memory解决方法
前言
- 今天在运行代码的时候出现了cuda的一个报错，报错如下，意思就是超出内存了
解决
- 首先查看一下GPU使用情况，命令如下：nvidia-smi
- 看输出的第二列（Memory_Usage）查看各个GPU使用情况
- 找到剩余内存较大的GPU，然后代码中输入如下代码
import os
import torch
os.environ['CUDA_VISIBLE_DEVICES'] = '2,6' # 代表可以使用第二个和第六个GPU
device=torch.device("cuda:0") # cuda:0代表使用的是第二个GPU，cuda:1代表使用的是第六个GPU(按照索引取值)
- 1
- 2
- 3
- 4
- 5
总结
- 遇到cuda内存问题，首先查看那个GPU可以使用，然后设置环境参数CUDA_VISIBLE_DEVICES即可
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/263573
推荐阅读
相关标签
