diff --git a/.gitignore b/.gitignore index fa05ad8b487305902cc0744fb9c194ddd937128d..5ed641567d03bbc77f0c409b17ad8126737ad189 100644 --- a/.gitignore +++ b/.gitignore @@ -26,7 +26,7 @@ _work/ .pytest_cache/ results*/ *_version.py -.pixi .venv +.pixi pixi.toml pixi.lock diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000000000000000000000000000000000000..a1eb1c81fda268121a3d029d4fc9eae5dbb812de --- /dev/null +++ b/Dockerfile @@ -0,0 +1,10 @@ +FROM mambaorg/micromamba:focal-cuda-12.3.1 + +USER root +RUN apt-get update && apt-get install -y --no-install-recommends nvidia-utils-545 && rm -rf /var/lib/apt/lists/* +USER $MAMBA_USER + +COPY --chown=$MAMBA_USER:$MAMBA_USER dist /tmp/dist + +ENV CONDA_OVERRIDE_CUDA=12.0 +RUN micromamba install --name base --yes --override-channels -c /tmp/dist -c conda-forge python=3.11 mednet 'pytorch=*=cuda120*' && micromamba clean --all --yes && rm -rf /tmp/dist