关于device命令的解释

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '4' # except is str not int
import torch
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 

关于以上命令解释

os.environ['CUDA_VISIBLE_DEVICES'] = '4'

表示程序在运行的过程中，只能看到第4块卡。编号从0 开始，即此时第4块卡对应的编号是0.

os.environ['CUDA_VISIBLE_DEVICES'] = '2,1，4'

表示程序在运行的过程中，只能看到第2,1，4块卡。编号从0 开始，即此时,2块卡对应的编码是0， 第1块卡对应的编号是1， 第4块卡对应的编号是2.

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 

“cuda:0” 表示使用第0块卡。