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
|
|
|
|
| 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
|
|
|
|
| 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
|