diff --git a/Dockerfile b/Dockerfile index 35d4be7..07f129f 100644 --- a/Dockerfile +++ b/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