fixed bug

Author: hephaex

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

@@ -8,14 +8,14 @@ ENV TORCHTEXT_VERSION=0.10.0
 ENV TENSORBOARDX_VERSION=2.1
 
 RUN python3 -m pip uninstall -y torch && \
-    python3 -m pip install -U setuptools pip && \
+    python3 -m pip install -U pip==21.1.2 setuptools==57.0.0 && \
     python3 -m pip install --no-cache-dir \
-        https://download.pytorch.org/whl/cu111/torch-${PYTORCH_VERSION}%2Bcu111-cp38-cp38-linux_x86_64.whl \
-        https://download.pytorch.org/whl/cu111/torchvision-${TORCHVISION_VERSION}%2Bcu111-cp38-cp38-linux_x86_64.whl \
-        https://download.pytorch.org/whl/torchaudio-${TORCHAUDIO_VERSION}-cp38-cp38-linux_x86_64.whl \
-	https://download.pytorch.org/whl/cu111/torchcsprng-0.2.1%2Bcu111-cp38-cp38-linux_x86_64.whl \
-	https://download.pytorch.org/whl/torchserve-0.3.0-py2.py3-none-any.whl \
-        https://download.pytorch.org/whl/torchtext-${TORCHTEXT_VERSION}-cp38-cp38-linux_x86_64.whl \
+        http://download.pytorch.org/whl/cu111/torch-${PYTORCH_VERSION}%2Bcu111-cp38-cp38-linux_x86_64.whl \
+        http://download.pytorch.org/whl/cu111/torchvision-${TORCHVISION_VERSION}%2Bcu111-cp38-cp38-linux_x86_64.whl \
+	torchaudio==${TORCHAUDIO_VERSION} \
+	torchserve \
+        torchtext==${TORCHTEXT_VERSION} \
+#	torchcsprng \
 	jsonargparse \
 	torchmetrics \
 	pyDeprecate \
@@ -24,8 +24,8 @@ RUN python3 -m pip uninstall -y torch && \
 	pretrainedmodels \
 	cnn-finetune \
 	keras4torch \
-	pytorch-lightning && \
-    python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}
+	pytorch-lightning 
+#    python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}
 
 # torch2trt PyTorch to TensorRT converter which utilizes the TensorRT Python API
 WORKDIR /tmp