PyTorch's DataParallel is only using one GPU

后端 未结 0 728
一整个雨季
一整个雨季 2020-12-01 11:55

I\'m trying to use two GPU\'s for training a model in PyTorch. I\'m using torch.nn.DataParallel but for some reason nvidia-smi is saying that I\'m

相关标签:
回答
  • 消灭零回复
提交回复
热议问题