Pull in updated from clou-ml-examples (#148)

Author: jacobtomlinson

source/examples/rapids-sagemaker-higgs/Dockerfile

@@ -2,9 +2,11 @@ ARG RAPIDS_IMAGE
 
 FROM $RAPIDS_IMAGE as rapids
 
-RUN apt-get update && apt-get install -y --no-install-recommends build-essential
-
-RUN source activate rapids && pip install sagemaker-training
+# add sagemaker-training-toolkit [ requires build tools ], flask [ serving ], and dask-ml
+RUN apt-get update && apt-get install -y --no-install-recommends build-essential \
+    && source activate rapids \
+    && pip3 install sagemaker-training cupy-cuda115 flask \
+    && pip3 install --upgrade protobuf
 
 # Copies the training code inside the container
 COPY rapids-higgs.py /opt/ml/code/rapids-higgs.py