#Cuda out of memory 问题解决

  1. RuntimeError:CUDA out of memory.Tried to allocate 256.00 MiB (GPU0;23.70GiBtotal capacity;844.99 MiB already allocated;186.44 MiB free:868.00 MiBreserved in total by PyTorch)If reserved memory is >allocated memory trysettingmax split size mb to avoid fragmentation. See documentation for MemoryManagement and PYTORCH CUDA ALLOC CONF
    错误截图

错误原因1：有占内存进程在运行

解决方案：
Nvidia-smi查看占内存的进程号
kill -KILL 进程号

错误原因2：batch size太大，尝试调小

# 深度学习 # pytorch # 人工智能
Logo
DAMO开发者矩阵

DAMO开发者矩阵，由阿里巴巴达摩院和中国互联网协会联合发起，致力于探讨最前沿的技术趋势与应用成果，搭建高质量的交流与分享平台，推动技术创新与产业应用链接，围绕“人工智能与新型计算”构建开放共享的开发者生态。

更多推荐

cover

Clawdbot一夜爆红，凭什么？

avatar DAMO开发者矩阵
cover

DREAMGEN：通过视频世界模型解锁机器人学习中的泛化性

avatar DAMO开发者矩阵
cover

一种用于机器人操作的通用单样本多模态主动感知框架：学习预测最优视角

avatar DAMO开发者矩阵