pytorch 指定GPU訓練

# 1：
torch.cuda.set_device(1)

# 2：
device = torch.device("cuda:1")

# 3：（官方推薦）
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

（同時調用兩塊GPU的話）
os.environ["CUDA_VISIBLE_DEVICES"] = '1,2'