os.environ[‘CUDA_VISIBLE_DEVICES‘]指定GPU无法生效解决方案

解决方法一：
将语句放在 import torch 之前

os.environ['CUDA_VISIBLE_DEVICES'] = args.gpu_devices
import torch

解决方法二：
命令行运行时指定

CUDA_VISIBLE_DEVICES=x python  xxx.py