RuntimeError: CUDA error: out of memory的解决【实测成功】

仅作为记录，大佬请跳过。

背景

博主使用linux服务器运行MIL_train.py程序时，出现RuntimeError: CUDA error: out of memory的错误（之前运行这个python木有问题）

解决

在MIL_train.py文件里加入：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

即可。

参考

感谢大佬博主文章：传送门

原因

服务器的默认gpu显卡正在使用，需要使用gpu的另一个空闲的显卡；linux里查看gpu的显卡使用状态用代码：

nvidia-smi

运行成功后：