JarrentWu1031/CCPL

GPU to use

arrrrr3186 opened this issue · 1 comments

arrrrr3186 commented

Hi,

I trying to execute train.py, and scripts is asking for GPU parameter,
as I am using Nvidia, so which GPU i should mention.

JarrentWu1031 commented

Hi,

I trying to execute train.py, and scripts is asking for GPU parameter, as I am using Nvidia, so which GPU i should mention.

Maybe you could just specify --gpu 0 as the default setting.