仅作为记录,大佬请跳过。
背景
博主使用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
运行成功后: