测试网络时，导入模型时，报错“out of memory”

device = torch.device(“cuda:0” if torch.cuda.is_available() else “cpu”)

在跑程序时使用：CUDA_VISIBLE_DEVICES = 3 python test.py

结果报错：“out of memory"

原因：服务器多卡时，可能因为torch.devices(“cuda:0” if torch.cuda.is_available() else “cpu”)默认是0号卡，而0号卡却已经满存

解决方法：改成torch.devices(“cuda:3” if torch.cuda.is_available()else “cpu”)