我是在AMD的显卡上遇到的。如果你看了别人
torch.multiprocessing.set_start_method('spawn')
的方式不行的话,可以把每一处
.to(device)
去掉,去掉.to(device)即可
CUDA_TENSORS_ON_WORKER = True
if not CUDA_TENSORS_ON_WORKER:
x = to_torch(x).to(device)
else:
x = to_torch(x)
感谢BJL同学的指导