Update Dockerfile.1.7-py38-cuda11.1

Author: hephaex

python-pytorch/Dockerfile.1.7-py38-cuda11.1

@@ -5,7 +5,6 @@ ENV TORCHVISION_VERSION=0.8.2
 ENV TORCHAUDIO_VERSION=0.7.2
 ENV TORCHTEXT_VERSION=0.8.1
 ENV TENSORBOARDX_VERSION=2.1
-ENV MXNET_VERSION=1.6.0
 
 RUN python3 -m pip uninstall -y torch && \
     python3 -m pip install --no-cache-dir \
@@ -33,8 +32,8 @@ RUN ldconfig /usr/local/cuda/targets/x86_64-linux/lib/stubs && \
     ldconfig
 
 RUN python3 -m pip install --no-cache-dir \
-    	    mpi4py==3.0.3 \
-            mlflow==1.12.1 \
+    	mpi4py==3.0.3 \
+        mlflow==1.12.1 \
 	    nni==1.9 \
 	    scikit-nni==0.2.1