Below you can find a small example showcasing this: cuda = torch.device('cuda') # Default CUDA device cuda0 = torch.device('cuda:0') cuda2 ...
確定! 回上一頁