pytorch或者tensorflow指定显卡运行

方式一：在代码中指定

在使用pytorch或者tensorflow的所有代码的首行添加如下代码：

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = '你想要的显卡号（默认从0开始）'

方式二：在命令中指定

CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES=显卡号 python xxx.py