fixed build error

hephaex · hephaex · commit 2c8a1ba2f779 · 2021-03-31T16:55:15.000+09:00
diff --git a/python-pytorch/Dockerfile.1.8-py38-cuda11.1 b/python-pytorch/Dockerfile.1.8-py38-cuda11.1
@@ -39,6 +39,7 @@ RUN python3 -m pip install --no-cache-dir \
     	mpi4py==3.0.3 \
         mlflow==1.15.0 \
 	nni==2.1 \
+	tensorboard-plugin-wit \
 	scikit-nni==0.2.1
 
 RUN apt autoclean && \
