torch分布式训练报错

报错:
ValueError: Error initializing torch.distributed using env:// rendezvous: environment variable RANK expected, but not set
解决方法: python -m torch.distributed.launch --nproc_per_node=gpu个数 tools/train.py -f configs/damoyolo_tinynasL25_S.py

如果想用vscode或者用pycharm调试这个代码,需要这么设置即可:

import os
import torch
os.environ['MASTER_ADDR'] = 'localhost'
os.environ['MASTER_PORT'] = '12355'
os.environ['RANK'] = '0'
os.environ["CUDA_VISIBLE_DEVICES"] = "7"
os.environ['WORLD_SIZE'] = '1'

CUDA_VISIBLE_DEVICES这个变量设置成自己想用的gpu序号即可。

你可能感兴趣的:(人工智能,pytorch)