weixmath / view-GCN

Pytorch code for view-GCN (CVPR2020)
77 stars 17 forks source link

How to enable multi GPU training? #11

Closed piseabhijeet closed 3 years ago

piseabhijeet commented 3 years ago

Hi @weixmath

How do i enable multi-GPU training? I always end up with CUDA ALLOCATION error.

weixmath commented 3 years ago

I trained this model on 12G 1080Ti and i think the training is fast enough. Maybe you can modify it by yourself.

auniquesun commented 1 year ago

@piseabhijeet @weixmath Hi, the multi-GPU implementation is available here !