Skip to content

Instantly share code, notes, and snippets.

@Palpatineli

Palpatineli/def

Created January 29, 2025 15:05
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save Palpatineli/d55cb422e86122e2a59d24166060ce89 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save Palpatineli/d55cb422e86122e2a59d24166060ce89 to your computer and use it in GitHub Desktop.
Download ZIP
minimal singulariy
Raw
def
BootStrap: docker
From: pytorch/pytorch:2.5.1-cuda11.8-cudnn9-devel
%files
main.py /app/main.py
%environment
export nnUNet_raw="/mnt/data/nnUNet_raw"
export nnUNet_preprocessed="/mnt/data/nnUNet_preprocessed"
export nnUNet_results="/mnt/data/nnUNet_results"
export CUDA_VISIBLE_DEVICES="0"
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
%post
apt-get update && apt-get install -y git
apt-get install -y g++
mkdir -p /app
cd /app
git clone https://github.com/Palpatineli/nnUNet.git
cd /app/nnUNet
pip install -e .
%startscript
python -u /app/main.py
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment