Jupyter notebook 在pytorch中，如何将第二个gpu设置为juptyer笔记本中的默认gpu？

我有两个gpu，想打开第二个juptyer笔记本，并确保其中的所有内容仅在第二个gpu而不是第一个gpu上运行。理想情况下，我希望在开始时运行一个单元格，而不是在多个位置传递device=1。

这种情况下的答案是调用set device

import torch
torch.cuda.set_device(1) 

然而，文件不鼓励这样做。根据签入（）上的github注释，我的场景是唯一合理的用例。首选方法是设置一个环境变量，在笔记本电脑中也可以：

import os
os.environ['CUDA_VISIBLE_DEVICES']='1'