search

charlesq34 / frustum-pointnets

Frustum PointNets for 3D Object Detection from RGB-D Data
Apache License 2.0
1.59k stars 538 forks source link

How to train with 2 GPUs? #79

Open liqisa opened 5 years ago

liqisa commented 5 years ago

I have two 1080ti.How can I use them in training this model. Thx

RollingIsland commented 5 years ago

Change the value after --gpu option as 1 in "command_train_v1(or v2).sh"

liqisa commented 5 years ago

Change the value after --gpu option as 1 in "command_train_v1(or v2).sh"

Thanks for your help. But I mean that how can i use the 2 gpus working parallely. "--gpu=1" denotes that the working gpu will be number 1,not number 0 and 1 ,right?

RollingIsland commented 5 years ago

Well..., I'm not so sure, I tried it on 2 Titan Xp and the commandline showed 2 gpus, but I don't know whether they worked parallely. You can try it first.

pouyaramezani commented 4 years ago

I have tried. It doesn't work parallel on 2 gpus with --gpu 1. It just exploit the gpu number 1. Any idea for this issue?

kwea123 commented 4 years ago

The code doesn't support multigpu training. So impossible.