在PyTorch中,可以通过以下步骤将第二个GPU设置为Jupyter笔记本中的默认GPU:
import os
import torch
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
这里,CUDA_DEVICE_ORDER
环境变量用于指定GPU设备的顺序,PCI_BUS_ID
表示按照PCI总线ID的顺序进行排序。CUDA_VISIBLE_DEVICES
环境变量用于指定可见的GPU设备索引,这里设置为1表示只使用第二个GPU。
注意:如果你的机器上有更多的GPU设备,你可以根据需要设置CUDA_VISIBLE_DEVICES
环境变量的值,以选择要使用的GPU设备。
这样,你就成功将第二个GPU设置为Jupyter笔记本中的默认GPU了。在后续的代码中,PyTorch将使用该GPU进行计算和训练。
领取专属 10元无门槛券
手把手带您无忧上云