- pytorch默认使用gpu编号为device:0的设备，可以使用
torch.nn.DataParallel(model, device_ids=[0, 1])torch.nn.
阅读背景：
pytorch关于多块gpu使用总结，报错AssertionError: Invalid device id_kongkongqixi的博客
来源：互联网
分享到：
