RuntimeError: all tensors must be on devices[0]

问题说明

跑郑哲东的AICITY2020的那个代码报错，查找之后发现问题是跑的代码多GPU，但指定的是单GPU

问题解决

在test2020.py中126行和129行的代码注释掉换成下面的代码

# set gpu ids
# if len(gpu_ids)>0:
# torch.cuda.set_device(gpu_ids[0])
# cudnn.benchmark = True
os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu_ids[0])