RuntimeError: CUDA error (10): invalid device ordinal
两张卡跑的train,在本机一张卡上test不了结果很简单,直接将model.module拿出来就行了因为之前双卡用的paralleldef__init__(self,module,device_ids=None,output_device=None,dim=0):super(DataParallel,self).__init__()ifnottorch.cuda.is_available():s