首页 > 其他 > 详细

解决RuntimeError: CUDA error: out of memory

时间:2021-03-11 22:33:07      阅读:1189      评论:0      收藏:0      [点我收藏+]

解决RuntimeError: CUDA error: out of memory

跑代码时出现这个错误,输入以下命令

$ nvidia-smi -l

技术分享图片

原来是卡0内存爆了,于是指定相对空闲的2卡

CUDA_VISIBLE_DEVICES=2 python my_script.py

可以正常运行

解决RuntimeError: CUDA error: out of memory

原文:https://www.cnblogs.com/Tony100K/p/14520567.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!