参考https://www.codeleading.com/article/37445832417/
在main文件里面import的时候写入
import torch.distributed as dist dist.init_process_group('gloo', init_method='file:///tmp/somefile', rank=0, world_size=1)
分布式训练就变成单卡了
参考https://www.codeleading.com/article/37445832417/
在main文件里面import的时候写入
import torch.distributed as dist dist.init_process_group('gloo', init_method='file:///tmp/somefile', rank=0, world_size=1)
分布式训练就变成单卡了