add bagua installation in dockerfile

Author: wangraying

dockers/base-cuda/Dockerfile

@@ -114,6 +114,10 @@ RUN \
     # install DeepSpeed
     pip install deepspeed==0.5.4
 
+RUN \
+    # install Bagua
+    curl -Ls https://raw.githubusercontent.com/BaguaSys/bagua/master/install_master.sh | bash
+
 RUN \
     # Show what we have
     pip --version && \