记录CUDA error: out of memory的成功解决
惊讶地发现:我的torch版本确实和作者预训练的不一致!于是重装了1.4.0的古早torch,激动地去 python main.py 时,这error甚至一点没变化……NVIDIA-smi后显示,运行前没有GPU被占用,方法1pass!os.environ 可以获取有关系统的各种信息,此语句可以指定所要使用的显卡,没有此句默认为0。希望可以帮到大家~也提醒自己长个教训,思路不要被局限住,也要努力拓宽自己的视野,不要人云亦云~我去交流群求助,大家纷纷说是显存的问题,于是我又smi了一下:。
原创
2023-06-05 20:27:48 ·
2424 阅读 ·
3 评论