Example 1: pytorch check gpu In [1]: import torch In [2]: torch.cuda.current_device() Out[2]: 0 In [3]: torch.cuda.device(0) Out[3]:
確定! 回上一頁