python 中指定GPU乱序解决方案

import os建议立即紧跟着
gpu的指定，不然会导致python中指定的GPU和nvidia-smi中的GPU序号不匹配

import os
os.environ['CUDA_DEVICE_ORDER'] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "4,5,6,7"
 
import torch
1
2
3
4
5