Add docker image with Cuda support.

Author: matthid

.gitattributes

@@ -0,0 +1,5 @@
+# Ensure all C and PHP files use LF.
+*.py         eol=lf
+*.cs         eol=lf
+*.sh         eol=lf
+*Dockerfile  eol=lf

entrypoint.sh

@@ -0,0 +1,4 @@
+#!/bin/env bash
+
+echo "my entry point"
+python -m myproject.hello

nix/build_container.nix

@@ -0,0 +1,46 @@
+{
+    project_dependencies ? import ./dependencies.nix {}
+}:
+let
+  pkgs = project_dependencies.pkgs;
+  lib = project_dependencies.lib;
+  cudatoolkit = project_dependencies.cudatoolkit;
+  project = import ./project.nix { project_dependencies = project_dependencies; };
+  entrypointScriptPath = ../entrypoint.sh; # Adjust the path as necessary
+  entrypointScript = pkgs.runCommand "entrypoint-script" {} ''
+    mkdir -p $out/bin
+    cp ${entrypointScriptPath} $out/bin/entrypoint
+    chmod +x $out/bin/entrypoint
+  '';
+in import ./docker/buildCudaLayeredImage.nix {
+  inherit cudatoolkit;
+  buildLayeredImage = pkgs.dockerTools.streamLayeredImage;
+  lib = pkgs.lib;
+  maxLayers = 2;
+  name = "project_nix";
+  tag = "latest";
+
+  contents = [
+    pkgs.coreutils
+    pkgs.findutils
+    pkgs.gnugrep
+    pkgs.gnused
+    pkgs.gawk
+    pkgs.bashInteractive
+    pkgs.which
+    pkgs.file
+    pkgs.binutils
+    pkgs.diffutils
+    pkgs.less
+    pkgs.gzip
+    pkgs.btar
+    pkgs.nano
+    (pkgs.python311.withPackages (ps: [
+      project
+    ]))
+    entrypointScript
+  ];
+  config = {
+    Entrypoint = ["${entrypointScript}/bin/entrypoint"];
+  };
+}

nix/docker/buildCudaLayeredImage.nix

@@ -0,0 +1,47 @@
+# https://sebastian-staffa.eu/posts/nvidia-docker-with-nix/
+# https://github.com/Staff-d/nix-cuda-docker-example
+{  cudatoolkit
+,  buildLayeredImage
+,  lib
+,  name
+,  tag ? null
+,  fromImage ? null
+,  contents ? null
+,  config ? {Env = [];}
+,  extraCommands ? ""
+,  maxLayers ? 2
+,  fakeRootCommands ? ""
+,  enableFakechroot ? false
+,  created ? "2024-03-08T00:00:01Z"
+,  includeStorePaths ? true
+}:
+
+let
+
+  # cut the patch version from the version string
+  cutVersion = with lib; versionString:
+    builtins.concatStringsSep "."
+      (take 3 (builtins.splitVersion versionString )
+    );
+
+  cudaVersionString = "CUDA_VERSION=" + (cutVersion cudatoolkit.version);
+
+  cudaEnv = [
+    "${cudaVersionString}"
+    "NVIDIA_VISIBLE_DEVICES=all"
+    "NVIDIA_DRIVER_CAPABILITIES=all"
+
+    "LD_LIBRARY_PATH=/usr/lib64/"
+  ];
+
+  cudaConfig = config // {Env = cudaEnv;};
+
+in buildLayeredImage {
+  inherit name tag fromImage
+    contents extraCommands
+    maxLayers
+    fakeRootCommands enableFakechroot
+    created includeStorePaths;
+
+  config = cudaConfig;
+}