update packages

Author: hephaex

python-pytorch/Dockerfile.1.11-py38-cuda11.3

@@ -8,16 +8,19 @@ ENV TORCHTEXT_VERSION=0.10.0
 ENV TENSORBOARDX_VERSION=2.1
 
 RUN python3 -m pip uninstall -y torch && \
-    python3 -m pip install -U pip==21.1.2 setuptools==57.0.0 && \
-    python3 -m pip install --no-cache-dir \
+    python3 -m pip install -U pip setuptools 
+RUN python3 -m pip install --no-cache-dir \
         http://download.pytorch.org/whl/cu113/torch-1.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl \
-        http://download.pytorch.org/whl/cu102/torchvision-0.12.0%2Bcu102-cp38-cp38-linux_x86_64.whl \
+	keras4torch==1.2.4 \
+	pytorch-lightning==1.3.4 \
+	torch-model-archiver==0.3.0 \
+	torch-tb-profiler==0.4.0 \
 	http://download.pytorch.org/whl/cu113/torchaudio-0.11.0%2Bcu113-cp38-cp38-linux_x86_64.whl \
-	http://download.pytorch.org/whl/torchserve-0.3.0-py2.py3-none-any.whl \
-        http://download.pytorch.org/whl/torchtext-0.12.0-cp38-cp38-linux_x86_64.whl \
-	http://download.pytorch.org/whl/cu101/torchcsprng-0.2.1%2Bcu101-cp38-cp38-linux_x86_64.whl \
+	torchmetrics==0.7.2 \
+	torchserve==0.5.3 \
+        torchtext==0.12.0 \
+        http://download.pytorch.org/whl/cu102/torchvision-0.12.0%2Bcu102-cp38-cp38-linux_x86_64.whl \
 	http://download.pytorch.org/whl/torch_model_archiver-0.3.0-py2.py3-none-any.whl \
-	http://download.pytorch.org/whl/torch_tb_profiler-0.1.0-py3-none-any.whl \
 	jsonargparse \
 	torchmetrics \
 	pyDeprecate \
@@ -28,13 +31,6 @@ RUN python3 -m pip uninstall -y torch && \
 	keras4torch \
 	pytorch-lightning 
 
-# torch2trt PyTorch to TensorRT converter which utilizes the TensorRT Python API
-WORKDIR /tmp
-RUN git clone --branch=v0.3.0 https://github.com/NVIDIA-AI-IOT/torch2trt /tmp/torch2trt && \
-    cd /tmp/torch2trt && \
-    python3 setup.py install --plugins && \
-    rm -fr /tmp/torch2trt  
-
 RUN python3 -m pip install --extra-index-url \
        https://developer.download.nvidia.com/compute/redist --upgrade nvidia-dali-cuda110
 
@@ -52,7 +48,9 @@ RUN python3 -m pip install --no-cache-dir \
 	tensorboard-plugin-wit \
 	scikit-nni==0.2.1
 
-RUN apt autoclean && \
+RUN apt update && \
+    apt install ncurses-term && \
+    apt autoclean && \
     sed -i 's/source \/usr\/local\/nvm\/nvm.sh//' /etc/bash.bashrc && \
     ln -sf /usr/share/terminfo/x/xterm-color /usr/share/terminfo/x/xterm-256color && \
     rm -rf /var/lib/apt/lists/* && \