pytorch-lightning

24ffde15 - Add bagua support for CUDA 11.6 images (#14529)

Add bagua support for CUDA 11.6 images (#14529) * Add support for bagua-cuda116 * Remove bagua-cuda115 from installation Co-authored-by: Akihiro Nitta <nitta@akihironitta.com>