diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py index 071be073ba756..da0c16d0e4f52 100644 --- a/pytorch_lightning/trainer/evaluation_loop.py +++ b/pytorch_lightning/trainer/evaluation_loop.py @@ -427,7 +427,7 @@ def evaluation_forward(self, model, batch, batch_idx, dataloader_idx, test_mode: # set cuda device to root gpu # related to https://github.com/PyTorchLightning/pytorch-lightning/issues/958 - # Refer solution: https://github.com/pytorch/pytorch/issues/9871#issuecomment-408304190 + # Refer: https://github.com/pytorch/pytorch/issues/9871#issuecomment-408304190 root_device = torch.device("cuda", root_gpu) torch.cuda.set_device(root_device) else: