使用GPU跑代码，如何指定显卡

1.选择使用GPU还是CPU运行代码

# device = torch.device("cpu")# 使用cpu训练
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

2.服务器一般有多个卡，跑代码时指定显卡运行

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1' #0指第一张卡，1指第二张卡