Question about train the code with multiple GPUs

yinboc / liif

Learning Continuous Image Representation with Local Implicit Image Function, in CVPR 2021 (Oral)

https://yinboc.github.io/liif/

BSD 3-Clause "New" or "Revised" License

1.26k stars 145 forks source link

Question about train the code with multiple GPUs #42

Closed xuejiancai closed 3 years ago

xuejiancai commented 3 years ago

Dear author, There are two GPUs on my machine.But when I trained with --gpu 0,1, I found through nvidia-smi that only gpu 0 was working. So how to solve this problem? Thank you!