服务器上有3张卡,但只使用第二张和第三张卡 设置环境变量 export CUDA_VISIBLE_DEVICES=1,0 只想使用第一张卡 import torch torch.cuda.set_device(0)