深度学习小白 pytorch遇见RuntimeError: CUDA out of memory
还是小白的时候第一次自己摸索着跑实验用显卡，但是并不了解显卡也没有概念，一个轻薄商务本的独显卡（就2G显存，当时还不会看显存 也不懂）我以为可以用，，结果一直报这个错 CUDA out of memory。然后我以为是我哪里的设置写错 就按教程去改batchsize什么的改半天，但其实根本没用！因为2G显存什么实验都不能跑！所以如果大家是这个轻薄本什么的独显的显存一般都不够的，一些游戏本可能会有3
·
还是小白的时候第一次自己摸索着跑实验用显卡，但是并不了解显卡也没有概念，一个轻薄商务本的独显卡（就2G显存，当时还不会看显存 也不懂）我以为可以用，，结果一直报这个错 CUDA out of memory。然后我以为是我哪里的设置写错 就按教程去改batchsize什么的改半天，但其实根本没用！！因为2G显存什么实验都不能跑！！！！太天真
所以如果大家是这个轻薄本什么的独显的显存一般都不够的，一些游戏本可能会有30或40系列的显卡就可以用 。我后来是配了3060， 12G显存，跑实验勉强可以用啦。
查看显存的方法：打开命令行输入 nvidia-smi(要装nvidia哦)
第二行第二列的2048MiB就是说我的这个卡显存就2G
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)