Test on one GPU

[mayingwuhu]

Hello,

Thank you very much for open-sourcing this code. I would like to try running it locally, but I only have one GPU, and I've encountered several issues while trying to install the Horovod library, making it impossible for me to proceed. Could you please let me know if there is a version for single GPU testing that doesn't require Horovod installation? Alternatively, could you provide instructions for setting up the environment directly through Conda?

Looking forward to your reply, Very thanks!