File size: 456 Bytes
b111f5d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
Bootstrap: docker
#From: python:latest
#From: python:3.11-slim
From: python:3.10
%post
# Create the app directory in /home within the container's filesystem
mkdir -p app
# Upgrade pip
pip install --no-cache-dir --upgrade pip
# Install dependencies
echo "Installing Python dependencies..."
pip install --no-cache-dir torch torchvision torchaudio lightning tensorboard pycocotools matplotlib
%runscript
# Run the given command
exec "$@" |