search

BNU-IVC / FastPoseGait

FastPoseGait is a user-friendly and flexible repository that aims to help researchers get started on pose-based gait recognition quickly.
64 stars 3 forks source link

torch.distributed.launch分布式训练只有一个gpu怎么解决 #14

Open carrot-wuziyu opened 2 hours ago

carrot-wuziyu commented 2 hours ago

ValueError: Error initializing torch.distributed using env:// rendezvous: environment variable RANK expected, but not set 电脑只有一个gpu，跑不通应该怎么解决

DreamShibei commented 2 hours ago

你好，可以把脚本中的启动命令设置为：CUDA_VISIBLE_DEVICES=0，--nproc_per_node=1