pytorch关于多块gpu使用总结，报错AssertionError: Invalid device id

———————————————— 版权声明：本文为CSDN博主「kongkongqixi」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。 原文链接：https://blog.csdn.net/kongkongqixi/article/details/100521590 ————————————————
pytorch默认使用gpu编号为device:0的设备，可以使用 torch.nn.DataParallel(model, device_ids=[0, 1]) 对模型进行制定gpu指定编号多gpu训练，必须要有编号为device:0的gpu，不然会报AssertionError: Invalid device id错误； 当gpu编号为device:0的设备被占用时，指定其他编号gpu使用 torch.nn.DataParallel(model, device_ids=

pytorch关于多块gpu使用总结，报错AssertionError: Invalid device id最先出现在Python成神之路。