摘要: 问题背景: 最近跑代码时发现报错CUDA out of memory,进入linux终端查看GPU使用情况(nvidia-smi),结果如下: 我用的GPU序号是0,但这块被人占用了,所以我可以用剩下的3号和4号。 解决方案: 在代码中更改GPU使用序号(修改/添加代码): 1 import os 阅读全文
posted @ 2021-09-17 09:36 achived 阅读(950) 评论(0) 推荐(0) 编辑