2020年5月12日
torch.load CUDA ERROR: out of memory
摘要： 在pytorch中load模型，出现CUDA ERROR:out of memory,基本有下面三个原因： CUDA显存不足 解决办法：换成显存更大的显卡 显卡被占用 当存在多个显卡时，pytorch默认使用0号显卡，当0号显卡被占用时，即使1号显卡空闲仍然会报错 解决办法：让其他显卡不可见，此时0 阅读全文
posted @ 2020-05-12 11:49 happy_fan 阅读(2924) 评论(0) 推荐(0) 编辑