device = torch.device('cuda:2') for GPU 2. Training on Multiple GPUs. To allow Pytorch to “see” all available GPUs, use ...
確定! 回上一頁