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

libtorch-bin: switch from CUDA 10.2 to 11.1 build

Browse Source
This commit is contained in:
Daniël de Kok 2021-03-08 09:30:36 +01:00
parent 374f9a7c3b
commit 70f5fe4223
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pkgs/development/libraries/science/math/libtorch/bin.nix
View File

@ -8,8 +8,8 @@
, fixDarwinDylibNames , fixDarwinDylibNames
, cudaSupport , cudaSupport
, cudatoolkit_10_2 , cudatoolkit_11_1
, cudnn_cudatoolkit_10_2 , cudnn_cudatoolkit_11_1
}: }:
let let
@ -109,8 +109,8 @@ in stdenv.mkDerivation {
passthru.tests.cmake = callPackage ./test { passthru.tests.cmake = callPackage ./test {
inherit cudaSupport; inherit cudaSupport;
cudatoolkit = cudatoolkit_10_2; cudatoolkit = cudatoolkit_11_1;
cudnn = cudnn_cudatoolkit_10_2; cudnn = cudnn_cudatoolkit_11_1;
}; };
meta = with lib; { meta = with lib; {

4
pkgs/development/libraries/science/math/libtorch/binary-hashes.nix
View File

@ -8,7 +8,7 @@ version: {
hash = "sha256-xBaNyI7eiQnSArHMITonrQQLZnZCZK/SWKOTWnxzdpc="; hash = "sha256-xBaNyI7eiQnSArHMITonrQQLZnZCZK/SWKOTWnxzdpc=";
}; };
x86_64-linux-cuda = { x86_64-linux-cuda = {
url = "https://download.pytorch.org/libtorch/cu102/libtorch-cxx11-abi-shared-with-deps-${version}.zip"; url = "https://download.pytorch.org/libtorch/cu111/libtorch-cxx11-abi-shared-with-deps-${version}%2Bcu111.zip";
hash = "sha256-rNEyE4+jfeX7cU0aNYd5b0pZGYT0PNPnDnS1PIsrMeM="; hash = "sha256-uQ7ptOuzowJ0JSPIvJHyNotBfpsqAnxpMDLq7Vl6L00=";
}; };
} }