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