search

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.
https://cg.cs.tsinghua.edu.cn/jittor/
Apache License 2.0
3.08k stars 311 forks source link

How to set multi GPUs while training? #34

Open zzjing83 opened 4 years ago

zzjing83 commented 4 years ago

Thanks for the excellent work! I am trying the SSD example and it works well on GPU. However, it is wokrs on one GPU but I can not find a way to set multi GPUS while training. Any suggestion is welcomed! Thanks again

MenghaoGuo commented 4 years ago

Thanks for your using Jittor. Now, we are developing the multi GPUs training feature and It is expected to be supported in the next version.

zzjing83 commented 4 years ago

Thanks! Looking forward your great job. Make Jittor Greater Again~