apptainer_def_files/py311_torch_alpine.def

Bootstrap: docker
From: alpine:latest

%environment
  export PATH="$PATH:/usr/local/bin"

%post
  mkdir /app

  # Install Python and pip
  apk add --update --no-cache python3

  python3 -m venv /app/venv

  # activate the virtual environment
  . /app/venv/bin/activate

  # Install dependencies
  echo "Installing Python dependencies..."
  # pip install numpy
  pip install torch torchvision torchaudio lightning tensorboard pycocotools matplotlib

%runscript
  . /app/venv/bin/activate
  # echo "This is the runscript"
  exec "$@"