Commit b139e331 authored by David Guibert's avatar David Guibert
Browse files

cudatoolkit: add 11.7

parent df653dcf
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -11,6 +11,7 @@ final: prev: let
    "11.4" = ./manifests/redistrib_11.4.4.json;
    "11.5" = ./manifests/redistrib_11.5.2.json;
    "11.6" = ./manifests/redistrib_11.6.2.json;
    "11.7" = ./manifests/redistrib_11.7.0.json;
  };

  # Function to build a single cudatoolkit redist package
+6 −0
Original line number Diff line number Diff line
@@ -59,3 +59,9 @@ version = "11.6.1"
url = "https://developer.download.nvidia.com/compute/cuda/11.6.1/local_installers/cuda_11.6.1_510.47.03_linux.run"
sha256 = "sha256-qyGa/OALdCABEyaYZvv/derQN7z8I1UagzjCaEyYTX4="
gcc = "gcc11"

["11.7"]
version = "11.7.0"
url = "https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run"
sha256 = "sha256-CH/fy7ofeVQ7H3jkOo39rF9tskLQQt3oIOFtwYWJLyY="
gcc = "gcc11"
+2 −1
Original line number Diff line number Diff line
@@ -94,7 +94,7 @@ final: prev: let
        fullVersion = "8.3.2.44";
        hash = "sha256-VQCVPAjF5dHd3P2iNPnvvdzb5DpTsm3AqCxyP6FwxFc=";
        url = "${urlPrefix}/v${majorMinorPatch fullVersion}/local_installers/${fileVersion}/cudnn-linux-x86_64-${fullVersion}_cuda${fileVersion}-archive.tar.xz";
        supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.4" "11.5" "11.6" ];
        supportedCudaVersions = [ "11.0" "11.1" "11.2" "11.3" "11.4" "11.4" "11.5" "11.6" "11.7" ];
      }
    ];
  };
@@ -111,6 +111,7 @@ final: prev: let
    "11.4" = "8.3.2";
    "11.5" = "8.3.2";
    "11.6" = "8.3.2";
    "11.7" = "8.3.2";
  }.${cudaVersion};

in cuDnnPackages
+3 −0
Original line number Diff line number Diff line
@@ -11,6 +11,9 @@ final: prev: let
    "11.4" = "082dkk5y34wyvjgj2p5j1d00rk8xaxb9z0mhvz16bd469r1bw2qk";
    "11.5" = "sha256-AKRZbke0K59lakhTi8dX2cR2aBuWPZkiQxyKaZTvHrI=";
    "11.6" = "sha256-AsLNmAplfuQbXg9zt09tXAuFJ524EtTYsQuUlV1tPkE=";
    # the tag 11.7 does not exists: see https://github.com/NVIDIA/cuda-samples/issues/128
    # maybe fixed by https://github.com/NVIDIA/cuda-samples/pull/133
    "11.7" = prev.lib.fakeSha256;
  }.${prev.cudaVersion};

in {
+2 −1
Original line number Diff line number Diff line
@@ -5192,7 +5192,8 @@ with pkgs;
  cudaPackages_11_4 = callPackage ./cuda-packages.nix { cudaVersion = "11.4"; };
  cudaPackages_11_5 = callPackage ./cuda-packages.nix { cudaVersion = "11.5"; };
  cudaPackages_11_6 = callPackage ./cuda-packages.nix { cudaVersion = "11.6"; };
  cudaPackages_11 = cudaPackages_11_6;
  cudaPackages_11_7 = callPackage ./cuda-packages.nix { cudaVersion = "11.7"; };
  cudaPackages_11 = cudaPackages_11_7;
  cudaPackages = recurseIntoAttrs cudaPackages_11;
  # TODO: move to alias