update pip

Author: hephaex

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

@@ -8,12 +8,13 @@ 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 --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.4.0-py2.py3-none-any.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 \
 	jsonargparse \
 	torchmetrics \
@@ -22,6 +23,7 @@ RUN python3 -m pip uninstall -y torch && \
 	PyYAML>=5.1 \
 	pretrainedmodels \
 	cnn-finetune \
+	keras4torch \
 	pytorch-lightning && \
     python3 -m pip install --no-cache-dir tensorboardX==${TENSORBOARDX_VERSION}