小白學GAN系列3——torch.device

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

  這段代碼一般寫在讀取數據之前，torch.device代表將torch.Tensor分配到的設備的對象。torch.device包含一個設備類型（‘cpu’或‘cuda’）和可選的設備序號。如果設備序號不存在，則為當前設備。如：torch.Tensor用設備構建‘cuda’的結果等同於‘cuda：X’，其中X是torch.cuda.current_device()的結果。