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
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