public/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #112648 from danieldk/cuda-update

Browse Source 
cudatoolkit_11, cudnn_cudatoolkit_11: 11.1 -> 11.2
This commit is contained in:
Bernardo Meurer
2021-02-14 05:44:24 +00:00
committed by GitHub
parent 4bca02dbe6 f1db117b9f
commit aaf0a68d81
3 changed files with 29 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/libraries/science/math/cudnn/default.nix
+15 -5
View File
@@ -1,4 +1,4 @@
{ 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"])) {
@@ -82,11 +82,21 @@ in rec {
cudnn_cudatoolkit_10 = cudnn_cudatoolkit_10_2;
cudnn_cudatoolkit_11_0 = generic rec {
version = "8.0.2";
version = "8.1.0";
cudatoolkit = cudatoolkit_11_0;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v8.0.2.39.tgz";
sha256 = "0ib3v3bgcdxarqapkxngw1nwl0c2a7zz392ns7w9ipcficl4cbv7";
# 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 = cudnn_cudatoolkit_11_0;
cudnn_cudatoolkit_11_1 = cudnn_cudatoolkit_11_0.override {
cudatoolkit = cudatoolkit_11_1;
};
cudnn_cudatoolkit_11_2 = cudnn_cudatoolkit_11_0.override {
cudatoolkit = cudatoolkit_11_2;
};
cudnn_cudatoolkit_11 = cudnn_cudatoolkit_11_2;
}