pytorch使用记录（六） 明确使用哪块GPU
可以用在多GPU的服务器上，明确使用的是哪块GPU，可以将信息打印出来，返回的是device的列表。具体用到的pytorch函数是torch.cuda.device_count()，返回可得到的GPU数量。接下来是具体函数代码，分为两部分，一个主函数get_proper_device明确使用GPU还是CPU，另外一个函数get_proper_cuda_device是明确具体使用哪块GPU。...
