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

Merge pull request #41259 from jyp/tf-1.7.1-bin

Browse Source 
python.pkgs.tensorflow: wheel version
This commit is contained in:
Frederik Rietdijk
2018-06-07 11:27:39 +02:00
committed by GitHub
parent e5d43df48c 82540b5e1a
commit 4dc7cc8cc7
8 changed files with 172 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,4 +1,4 @@
{ callPackage, cudatoolkit7, cudatoolkit75, cudatoolkit8, cudatoolkit9 }:
{ callPackage, cudatoolkit_7, cudatoolkit_7_5, cudatoolkit_8, cudatoolkit_9_0, cudatoolkit_9 }:
let
generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) {
@@ -8,37 +8,44 @@ let
in
{
cudnn_cudatoolkit7 = generic rec {
cudnn_cudatoolkit_7 = generic rec {
version = "4.0";
cudatoolkit = cudatoolkit7;
cudatoolkit = cudatoolkit_7;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v${version}-prod.tgz";
sha256 = "0zgr6qdbc29qw6sikhrh6diwwz7150rqc8a49f2qf37j2rvyyr2f";
};
cudnn_cudatoolkit75 = generic rec {
cudnn_cudatoolkit_7_5 = generic rec {
version = "6.0";
cudatoolkit = cudatoolkit75;
cudatoolkit = cudatoolkit_7_5;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v${version}.tgz";
sha256 = "0b68hv8pqcvh7z8xlgm4cxr9rfbjs0yvg1xj2n5ap4az1h3lp3an";
};
cudnn6_cudatoolkit8 = generic rec {
cudnn6_cudatoolkit_8 = generic rec {
version = "6.0";
cudatoolkit = cudatoolkit8;
cudatoolkit = cudatoolkit_8;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v${version}.tgz";
sha256 = "173zpgrk55ri8if7s5yngsc89ajd6hz4pss4cdxlv6lcyh5122cv";
};
cudnn_cudatoolkit8 = generic rec {
cudnn_cudatoolkit_8 = generic rec {
version = "7.0.5";
cudatoolkit = cudatoolkit8;
cudatoolkit = cudatoolkit_8;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.tgz";
sha256 = "9e0b31735918fe33a79c4b3e612143d33f48f61c095a3b993023cdab46f6d66e";
};
cudnn_cudatoolkit9 = generic rec {
cudnn_cudatoolkit_9_0 = generic rec {
version = "7.0.5";
cudatoolkit = cudatoolkit9;
cudatoolkit = cudatoolkit_9_0;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.tgz";
sha256 = "03mbv4m5lhwnc181xz8li067pjzzhxqbxgnrfc68dffm8xj0fghs";
};
cudnn_cudatoolkit_9 = generic rec {
version = "7.0.5";
cudatoolkit = cudatoolkit_9;
srcName = "cudnn-${cudatoolkit.majorVersion}-linux-x64-v7.tgz";
sha256 = "1rfmdd2v47p83fm3sfyvik31gci0q17qs6kjng6mvcsd6akmvb8y";
};

10
pkgs/development/libraries/science/math/nccl/default.nix
View File

@@ -1,4 +1,4 @@
{ callPackage, cudatoolkit8, cudatoolkit9 }:
{ callPackage, cudatoolkit_8, cudatoolkit_9 }:
let
generic = args: callPackage (import ./generic.nix (removeAttrs args ["cudatoolkit"])) {
@@ -8,16 +8,16 @@ let
in
{
nccl_cudatoolkit8 = generic rec {
nccl_cudatoolkit_8 = generic rec {
version = "2.1.4";
cudatoolkit = cudatoolkit8;
cudatoolkit = cudatoolkit_8;
srcName = "nccl_${version}-1+cuda${cudatoolkit.majorVersion}_x86_64.txz";
sha256 = "1lwwm8kdhna5m318yg304kl2gsz1jwhv4zv4gn8av2m57zh848zi";
};
nccl_cudatoolkit9 = generic rec {
nccl_cudatoolkit_9 = generic rec {
version = "2.1.4";
cudatoolkit = cudatoolkit9;
cudatoolkit = cudatoolkit_9;
srcName = "nccl_${version}-1+cuda${cudatoolkit.majorVersion}_x86_64.txz";
sha256 = "0pajmqzkacpszs63jh2hw2qqc49kj75kcf7r0ky8hdh560q8xn0p";
};