解决RuntimeError: CUDA error: invalid device ordinal

在程序中找到定义 device 的地方，把.cuda(output_device)中的属性值改为0，即意味着只有一个GPU训练模型

版权声明：本文为com_fang_bean原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。