今天第一次在服务器上跑代码 也是第一次接触cuda显存卡
碰到了标题问题 将解决办法记录下来
1.
首先去查看cuda显存占用情况:
nvidia-smi
显示所有GPU的当前信息状态(图源网络 不想再去截图啦 介意可删)
如图片中第二列: 0 Mib/7605 Mib 这提醒我们内存并没有占满
因此我就去看了看是不是代码问题还是图片输入分辨率问题
接着考虑第二种情况
2.
如果是训练中出现这个提示,可考虑将batch_size调小一些
如果在测试集中 在代码问题的上一行 加入with torch.no_grad():
if __name__ == '__main__':
dehaze_list=[]
gt_list=[]
test_list = glob.glob("data/SOTS/HR_hazy/*")
for image in test_list:
with torch.no_grad():
clean_image=dehaze_image(image)