Dockerfile

FROM nvidia/cuda:12.1.1-devel-ubuntu22.04

LABEL maintainer="jan.hoenig@fau.de"
LABEL version="0.2"
LABEL description="Custom docker image for pystencils-benchmark"

ARG DEBIAN_FRONTEND=noninteractive

RUN apt-get update

RUN apt-get install -y \
    python3 \
    python3-pip \
    tox \
    build-essential \
    clang \
    gcc \
    libomp-dev && \
    rm -rf /var/lib/apt/lists/* && \
    apt-get clean