热门标签
热门文章
- 1【错误解决方案】Error: module ‘cv2‘ has no attribute ‘SURF‘_attributeerror: module 'cv2' has no attribute 'sur
- 2091、Python如何实现页面分页_python 分页界面
- 3 前端开发之走进Vue.js
- 4sublime Text 配置Node.js_sublimetext node.js
- 5特征值分解（Eigen Value Decomposition，EVD）、奇异值分解（Singular Value Decomposition，SVD）原理、公式推导及应用
- 6git合并本地提交记录_git如何合并提交记录
- 73D Gaussian Splatting 应用场景及最新进展【附10篇前沿论文和代码】_3d gaussian splatting csdn
- 8USB转I2C芯片操作EEPROM--CH347应用
- 9如何取消eslint校验
- 10Unity发布WebGl注意事项_your browser does not support graphics api "webgl
当前位置: article > 正文
使用GPU处理数据时显卡内存不够（Keras、tensorflow、pytorch等的Out of Memory报错）_无论batch_size调多小,总提示gpu内存不够
作者：数据挖掘灵魂 | 2024-02-01 12:21:56
赞
踩
无论batch_size调多小,总提示gpu内存不够
使用GPU处理数据时显卡内存不够（Keras、tensorflow、pytorch等的Out of Memory报错）
在数据处理、机器学习时，如果数据量过大，或者Batch_size设置地太大，或者因为前一次使用GPU之后没有释放内存，导致GPU内存不足报错，keras报CUDA_ERROR_OUT_OF_MEMOR，pytorch报OOM错误等
-
查看Ubuntu缓存（每一秒刷新）
watch -n 1 free -m
-
查看Ubuntu显存（每一秒刷新）
watch -n 1 nvidia-smi
-
关闭之前使用了GPU的终端等
-
如果没有其它占用NVIDIA显卡内存的你之前运行的进程，那么就是数据太大，或者Batch_size太大，设置较小的Batch_size
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/article/detail/53763
推荐阅读
相关标签
