查看显卡的使用情况
首先我们在终端使用如下代码,查看我们GPU的使用情况,watch -n 1(代表1秒更新显示一次)
watch -n 1 nvidia-smi
如下是出现的界面
我们根据这个列表可以看到有两块GPU,编号为0的GPU正在被使用所以现在我们指定编号为 1 的GPU进行训练
指定特定GPU的使用
import os
os.environ['CUDA_DEVICE_ORDER'] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = ‘1’