Code: Select all
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
Code: Select all
device = torch.device("cuda:0,1" if torch.cuda.is_available() else "cpu")
Code: Select all
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
Code: Select all
device = torch.device("cuda:0,1" if torch.cuda.is_available() else "cpu")