在使用os指定显卡时,需要放置在import torch 之前,否则不生效
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '7'
os.environ['MASTER_PORT']='6007'
import torch
在使用os指定显卡时,需要放置在import torch 之前,否则不生效
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '7'
os.environ['MASTER_PORT']='6007'
import torch