在這裡要注意的是,使用者必須要手動去檢查是否有多個GPU,藉著呼叫函示 torch.cuda.device_count() 來得到GPU 的個數。 PyTorch 的 nn.DataParallel 將許多關於多線程的 ...
確定! 回上一頁