Masoud KA commited on
Commit
7435efd
·
1 Parent(s): ee2d00d

Update dependencies in py311_almalinux9.def

Browse files
apptainer_def_files/py311_almalinux9.def CHANGED
@@ -103,10 +103,12 @@ From: almalinux:9
103
  pip install --no-cache-dir numpy==1.26.4 opencv-python-headless==4.10.0.84 pillow==10.4.0 && \
104
  pip install --no-cache-dir torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121 && \
105
  pip install --no-cache-dir cuda-python==12.1.0 lightning==2.4.0 mlflow==2.16.2 torchmetrics==1.4.2 tensorboard==2.18.0 && \
106
- pip install --no-cache-dir pycocotools==2.0.8 matplotlib==3.9.2 iopath==0.1.10 boto3 ipykernel psutil && \
 
107
  pip install --no-cache-dir mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.4/index.html && \
108
  pip install --no-cache-dir xformers==0.0.28 --index-url https://download.pytorch.org/whl/cu121 && \
109
- pip install --no-cache-dir json-tricks==3.17.3 xtcocotools==1.14.3 kornia==0.7.3
 
110
 
111
  %environment
112
  # Set the environment variable to activate the virtual environment by default
 
103
  pip install --no-cache-dir numpy==1.26.4 opencv-python-headless==4.10.0.84 pillow==10.4.0 && \
104
  pip install --no-cache-dir torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/cu121 && \
105
  pip install --no-cache-dir cuda-python==12.1.0 lightning==2.4.0 mlflow==2.16.2 torchmetrics==1.4.2 tensorboard==2.18.0 && \
106
+ pip install --no-cache-dir pycocotools==2.0.8 pycocoevalcap==1.2 xtcocotools==1.14.3 && \
107
+ pip install --no-cache-dir matplotlib==3.9.2 iopath==0.1.10 boto3==1.35.34 jupyter==1.1.1 psutil json-tricks==3.17.3 h5py==3.12.1 tabulate==0.9.0 && \
108
  pip install --no-cache-dir mmcv==2.2.0 -f https://download.openmmlab.com/mmcv/dist/cu121/torch2.4/index.html && \
109
  pip install --no-cache-dir xformers==0.0.28 --index-url https://download.pytorch.org/whl/cu121 && \
110
+ pip install --no-cache-dir kornia==0.7.3 transformers==4.45.2 timm==1.0.11 && \
111
+ pip install --no-cache-dir tensorrt==10.5.0
112
 
113
  %environment
114
  # Set the environment variable to activate the virtual environment by default