no_cuda does not take effect in non distributed environment (#23795)

Signed-off-by: Wang, Yi <yi.a.wang@intel.com>
This commit is contained in:
Wang, Yi
2023-05-26 22:47:51 +08:00
committed by GitHub
parent d61d747627
commit b7b729b38d

View File

@@ -1684,7 +1684,9 @@ class TrainingArguments:
)
device = torch.device("mps")
self._n_gpu = 1
elif self.no_cuda:
device = torch.device("cpu")
self._n_gpu = 0
else:
# if n_gpu is > 1 we'll use nn.DataParallel.
# If you only want to use a specific subset of GPUs use `CUDA_VISIBLE_DEVICES=0`