Add devcontainer support with CUDA

Author: Dominik Falkner

README.md

@@ -13,6 +13,7 @@ A simple [Cookiecutter](https://github.com/cookiecutter/cookiecutter) template f
 ## Batteries Included
 
 - **Packaging and dependency management** with [Poetry](https://github.com/python-poetry/poetry)
+- **Ready to use CUDA DEVCONTAINER** with [devcontainers](https://containers.dev/)
 - **Task running** with [Poe the Poet](https://github.com/nat-n/poethepoet)
    - **Pre defined** tasks
    - **Code formatting** with [Black](https://github.com/psf/black), [isort](https://github.com/PyCQA/isort)

{{ cookiecutter.__package_name_kebab_case }}/.devcontainer/devcontainer.json

@@ -0,0 +1,35 @@
+// For format details, see https://aka.ms/devcontainer.json. For config options, see the
+// README at: https://github.com/devcontainers/templates/tree/main/src/docker-existing-dockerfile
+{
+	"name": "CUDA Container ",
+	"build": {
+		// Sets the run context to one level up instead of the .devcontainer folder.
+		"context": "..",
+		// Update the 'dockerFile' property if you aren't using the standard 'Dockerfile' filename.
+		"dockerfile": "../Dockerfile-cuda"
+	},
+	"features": {
+		"ghcr.io/devcontainers/features/git-lfs:1": {
+			"autoPull": true,
+			"version": "latest"
+		}
+	},
+	"runArgs": [
+		"--gpus=all"
+	],
+
+	// Features to add to the dev container. More info: https://containers.dev/features.
+	// "features": {},
+
+	// Use 'forwardPorts' to make a list of ports inside the container available locally.
+	// "forwardPorts": [],
+
+	// Uncomment the next line to run commands after the container is created.
+	// "postCreateCommand": "cat /etc/os-release",
+
+	// Configure tool-specific properties.
+	// "customizations": {},
+
+	// Uncomment to connect as an existing user other than the container default. More info: https://aka.ms/dev-containers-non-root.
+	// "remoteUser": "devcontainer"
+}

{{ cookiecutter.__package_name_kebab_case }}/Dockerfile-cuda

@@ -0,0 +1,65 @@
+# This dockerfile serves as a base for all testing images.
+# It contains the minimal dependencies and is therefore smaller.
+from nvidia/cuda:12.3.1-runtime-ubuntu22.04
+ARG PYTHON_VERSION=3.10
+ARG POETRY_VERSION=1.7.1
+
+ENV DEBIAN_FRONTEND noninteractive
+LABEL maintainer="Dominik Falkner <dominik.falkner@risc-software.at>"
+
+RUN apt-get update --allow-insecure-repositories && apt-get install -yq \ 
+    build-essential \
+    # provides nohup command
+    coreutils \
+    # editor
+    vim \
+    libssl-dev \
+    zlib1g-dev \
+    libbz2-dev \
+    libreadline-dev \
+    libsqlite3-dev \
+    curl \ 
+    libncursesw5-dev \
+    xz-utils \
+    tk-dev \
+    libxml2-dev \
+    libxmlsec1-dev \
+    libffi-dev \
+    liblzma-dev \
+    git \
+    python3-distutils \
+    && rm -rf /var/lib/apt/lists/*
+    
+ENV PYENV_ROOT $HOME/.pyenv
+ENV PATH $PYENV_ROOT/shims:$PYENV_ROOT/bin:$PATH
+
+# Install pyenv
+RUN set -ex \
+    && curl https://pyenv.run | bash \
+    && pyenv update \
+    && eval "$(pyenv init -)" && pyenv --version
+
+# Install python and set interpreter
+RUN pyenv install -s ${PYTHON_VERSION} && pyenv global ${PYTHON_VERSION}
+
+# Install and configure Poetry
+ENV POETRY_VERSION=${POETRY_VERSION} \
+    POETRY_HOME=/opt/poetry \
+    POETRY_VIRTUALENVS_CREATE=false
+RUN curl -sSL https://install.python-poetry.org | python -
+ENV PATH="${PATH}:${POETRY_HOME}/bin"
+
+# Setup directory structure
+RUN mkdir /repo
+WORKDIR /repo
+# set pythonpath for nvidia image
+ENV PYTHONPATH "${PYTHONPATH}:/repo"
+
+# Install dependencies (package itself is not installed)
+COPY pyproject.toml poetry.lock .
+RUN poetry install --all-extras --no-root
+
+# Copy the rest of the repository
+ADD . .
+
+ENTRYPOINT ["tail", "-f", "/dev/null"]