nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-10-02 04:50:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

{,cudnn_}cudatoolkit_11_{1,2}: Port over from master

Browse Source 
This fixes eval. Sorry I broke it!
This commit is contained in:
John Ericson
2021-03-18 17:58:45 -04:00
parent a3b3c488e8
commit 4cbbe3b0c0
3 changed files with 39 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
pkgs/development/compilers/cudatoolkit/default.nix
View File

@@ -146,5 +146,19 @@ in rec {
gcc = gcc9;
};
cudatoolkit_11_1 = common {
version = "11.1.1";
url = "https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run";
sha256 = "13yxv2fgvdnqqbwh1zb80x4xhyfkbajfkwyfpdg9493010kngbiy";
gcc = gcc9;
};
cudatoolkit_11_2 = common {
version = "11.2.1";
url = "https://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.run";
sha256 = "sha256-HamMuJfMX1inRFpKZspPaSaGdwbLOvWKZpzc2Nw9F8g=";
gcc = gcc9;
};
cudatoolkit_11 = cudatoolkit_11_0;
}

20
pkgs/development/libraries/science/math/cudnn/default.nix
View File

@@ -1,4 +1,9 @@
{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9_0, cudatoolkit_9_1, cudatoolkit_9_2, cudatoolkit_10_0, cudatoolkit_10_1, cudatoolkit_10_2, cudatoolkit_11_0 }:
{ callPackage
, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8
, cudatoolkit_9_0, cudatoolkit_9_1, cudatoolkit_9_2
, cudatoolkit_10_0, cudatoolkit_10_1, cudatoolkit_10_2
, cudatoolkit_11_0, cudatoolkit_11_1, cudatoolkit_11_2
}:
let
generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) {
@@ -88,5 +93,18 @@ in rec {
sha256 = "0ib3v3bgcdxarqapkxngw1nwl0c2a7zz392ns7w9ipcficl4cbv7";
};
cudnn_cudatoolkit_11_1 = generic rec {
version = "8.1.0";
cudatoolkit = cudatoolkit_11_0;
# 8.1.0 is compatible with CUDA 11.0, 11.1, and 11.2:
# https://docs.nvidia.com/deeplearning/cudnn/support-matrix/index.html#cudnn-cuda-hardware-versions
srcName = "cudnn-11.2-linux-x64-v8.1.0.77.tgz";
sha256 = "sha256-2+gvrwcdkbqbzwBIAUatM/RiSC3+5WyvRHnBuNq+Pss=";
};
cudnn_cudatoolkit_11_2 = cudnn_cudatoolkit_11_1.override {
cudatoolkit = cudatoolkit_11_2;
};
cudnn_cudatoolkit_11 = cudnn_cudatoolkit_11_0;
}

8
pkgs/top-level/all-packages.nix
View File

@@ -2947,7 +2947,9 @@ in
cudatoolkit_10_1
cudatoolkit_10_2
cudatoolkit_11
cudatoolkit_11_0;
cudatoolkit_11_0
cudatoolkit_11_1
cudatoolkit_11_2;
cudatoolkit = cudatoolkit_10;
@@ -2966,7 +2968,9 @@ in
cudnn_cudatoolkit_10_1
cudnn_cudatoolkit_10_2
cudnn_cudatoolkit_11
cudnn_cudatoolkit_11_0;
cudnn_cudatoolkit_11_0
cudnn_cudatoolkit_11_1
cudnn_cudatoolkit_11_2;
cudnn = cudnn_cudatoolkit_10;