PyTorch指定GPU进行训练

1 直接在终端中设定：

CUDA_VISIBLE_DEVICES=1 python main.py

2 python 代码中设定：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

审核编辑 黄昊