| Bootstrap: docker | |
| #From: python:latest | |
| From: python:3.11-slim | |
| %post | |
| mkdir -p /opt | |
| echo "Installing required packages..." | |
| apt-get update && apt-get install -y libc6 libhdf5-103-1 libhdf5-hl-100 liblz4-1 libreadline8 slurm-wlm-basic-plugins munge | |
| # Clean up | |
| apt-get clean | |
| #rm -rf /var/lib/apt/lists/* | |
| #rm -rf /tmp/* | |
| # Set up the virtual environment | |
| mkdir /app | |
| python -m venv /app/venv | |
| # activate the virtual environment | |
| . /app/venv/bin/activate | |
| # Install dependencies | |
| echo "Installing Python dependencies..." | |
| # pip install --no-cache-dir pycocotools | |
| pip install --no-cache-dir torch torchvision torchaudio lightning tensorboard pycocotools matplotlib | |
| %runscript | |
| . /app/venv/bin/activate | |
| # echo "This is the runscript" | |
| exec "$@" | |