libopenmpi-dev (#44)

Dockerfile

@@ -21,7 +21,9 @@ RUN apt -y update && apt -y upgrade && \
         libglib2.0-0 \
         libgtk2.0-dev \
         libjpeg-dev \
+        libmpich-dev \
         libopenexr-dev \
+        libopenmpi-dev \
         libpng-dev \
         libsm6 \
         libssl-dev \
@@ -49,7 +51,7 @@ RUN apt -y update && apt -y upgrade && \
 RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install --no-cache-dir \
         setuptools \
-        wheel==0.41.1 && \
+        wheel==0.41.2 && \
     python3 -m pip install --no-cache-dir \
         torch==2.0.1+cu118 \
         torchvision==0.15.2+cu118 \
@@ -59,91 +61,72 @@ RUN python3 -m pip install --upgrade pip && \
 
 RUN python3 -m pip install --upgrade pip && \
     python3 -m pip install --no-cache-dir \
+        accelerate==0.23.0 \
         albumentations==1.3.1 \
         bitsandbytes==0.41.1 \
-        boto3==1.28.20 \
-        clip-interrogator==0.6.0 \
-        ctranslate2==3.18.0 \
-        cupy-cuda11x==12.1.0 \
-        datasets==2.14.3 \
-        deepspeed==0.10.0 \
-        dicomsdl==0.109.2 \
+        boto3==1.28.50 \
+        ctranslate2==3.20.0 \
+        cupy-cuda11x==12.2.0 \
+        datasets==2.14.5 \
         e2cnn==0.2.3 \
         effdet==0.4.1 \
         einops==0.6.1 \
         ensemble-boxes==1.0.9 \
+        evaluate==0.4.0 \
         faiss-gpu==1.7.2 \
+        fire==0.5.0 \
         gdcm==1.1 \
-        gensim==4.3.1 \
+        gensim==4.3.2 \
         googletrans==3.0.0 \
         google-trans-new==1.1.9 \
         grad-cam==1.4.8 \
         h5py==3.9.0 \
         hydra-core==1.3.2 \
         imutils==0.5.4 \
-        itk==5.3.0 \
-        jsonschema==4.18.6 \
+        jsonschema==4.19.0 \
         kornia==0.7.0 \
         kornia-moons==0.2.6 \
-        lightgbm==4.0.0 \
+        langchain==0.0.294 \
+        lightgbm==4.1.0 \
         lightning==2.0.6 \
-        loguru==0.7.0 \
-        matplotlib==3.7.2 \
-        monai==1.2.0 \
+        loguru==0.7.2 \
+        matplotlib==3.8.0 \
+        mpi4py==3.1.4 \
         networkx==3.1 \
         nltk==3.8.1 \
         onnxruntime==1.15.1 \
         onnxsim==0.4.33 \
-        opencv-python==4.8.0.74 \
-        open-clip-torch==2.20.0 \
-        openmim==0.3.9 \
-        optuna==3.2.0 \
-        pandas==2.0.3 \
-        peft==0.4.0 \
-        polars==0.18.12 \
-        pycocotools==2.0.6 \
-        pydicom==2.4.2 \
+        opencv-python==4.8.0.76 \
+        optuna==3.3.0 \
+        pandas==2.1.0 \
+        peft==0.5.0 \
+        polars==0.19.3 \
+        pycocotools==2.0.7 \
         pylibjpeg==1.4.0 \
         pylibjpeg-libjpeg==1.3.4 \
         pylibjpeg-openjpeg==1.3.2 \
         pylibjpeg-rle==1.3.0 \
-        python-gdcm==3.0.22 \
         pytorch-metric-learning==2.3.0 \
         pyvips==2.2.1 \
+        PyYAML==6.0.1 \
         segmentation-models-pytorch==0.3.3 \
         sentencepiece==0.1.99 \
         sentence-transformers==2.2.2 \
         torchstain==1.3.0 \
-        tqdm==4.65.0 \
-        transformers==4.31.0 \
+        tqdm==4.66.1 \
+        transformers==4.33.2 \
         translate==3.6.1 \
         scikit-learn==1.3.0 \
         scikit-image==0.21.0 \
-        scipy==1.11.1 \
+        scipy==1.11.2 \
         seaborn==0.12.2 \
         warmup_scheduler==0.3 \
-        wandb==0.15.8 \
-        xformers==0.0.20 \
+        wandb==0.15.10 \
+        xformers==0.0.21 \
         yacs==0.1.8 && \
-    python3 -m pip install --no-cache-dir \
-        --extra-index-url https://developer.download.nvidia.com/compute/redist/nightly --upgrade nvidia-dali-nightly-cuda110 && \
-    python3 -m pip cache purge
-
-RUN mim install mmcv-full
-
-RUN python3 -m pip install --upgrade pip && \
-    python3 -m pip install --no-cache-dir \
-        mmpretrain==1.0.1 \
-        mmdet==3.1.0 \
-        mmsegmentation==1.1.1 && \
-    python3 -m pip cache purge
-
-RUN python3 -m pip install --upgrade pip && \
-    python3 -m pip install --no-cache-dir \
-        tensorrt==8.6.1 && \
     python3 -m pip cache purge
 
-RUN python3 -m pip install --no-cache-dir -U timm==0.9.5
+RUN python3 -m pip install --no-cache-dir -U timm==0.9.7
 RUN python3 -m pip install --no-cache-dir -U protobuf==3.20.2
 
 RUN python3 -m pip cache purge