search

WHUHLX / CATS

59 stars 16 forks source link

About multi-gpus training #4

Open isbrycee opened 3 years ago

isbrycee commented 3 years ago

Hi, I have set 'self.gpu = '0, 1,2,3,4,5,6,7'' in config file, but when I start to training, I found that only one gpu was used, how could I fix this issue? Thank you!

WHUHLX commented 3 years ago

The codes are used with singe gpu. If you want to use multiple gups, you can modify the codes with reference to https://towardsdatascience.com/distributed-neural-network-training-in-pytorch-5e766e2a9e62 for training with multiple gpus.