服务器共用的时候需要进行gpu分配,每个人用哪几块。? 方法1 在训练代码内加上: import os os.environ['CUDA_VISIBLE_DEVICES']='0,1' 方法2 在训练指令前面加上CUDA_VISIBLE_DEVICES=1,即: CUDA_VISIBLE_DEVICES=1 python train.py …… HOPE U ENJOY