pytorch使使用用单单个个GPU与与多多个个GPU进进行行训训练练与与测测试试的的方方法法

今天小编就为大家分享一篇pytorch使用单个GPU与多个GPU进行训练与测试的方法，具有很的参考价值，希

望对大家有所帮助。一起跟随小编过来看看吧

如如下下所所示示：：

device=torch.device(cuda:0iftorch.cuda.is_available()elsecpu)#第一行代码

model.to(device)#第二行代码

首先是上面两行代码放在读取数据之前。

mytesor=my_tesor.to(device)#第三行