Merge pull request #190913 from npatsakula/libtorch-bin-update

libtorch-bin: 1.10.0 -> 1.12.1
This commit is contained in:
Mario Rodas 2022-09-24 11:45:49 -05:00 committed by GitHub
commit 1b9ed9f801
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 1 deletions

View file

@ -17,7 +17,7 @@ let
# this derivation. However, we should ensure on version bumps # this derivation. However, we should ensure on version bumps
# that the CUDA toolkit for `passthru.tests` is still # that the CUDA toolkit for `passthru.tests` is still
# up-to-date. # up-to-date.
version = "1.10.0"; version = "1.12.1";
device = if cudaSupport then "cuda" else "cpu"; device = if cudaSupport then "cuda" else "cpu";
srcs = import ./binary-hashes.nix version; srcs = import ./binary-hashes.nix version;
unavailable = throw "libtorch is not available for this platform"; unavailable = throw "libtorch is not available for this platform";

View file

@ -16,4 +16,21 @@ version : builtins.getAttr version {
hash = "sha256-jPylK4j0V8SEQ8cZU+O22P7kQ28wanIB0GkBzRGyTj8="; hash = "sha256-jPylK4j0V8SEQ8cZU+O22P7kQ28wanIB0GkBzRGyTj8=";
}; };
}; };
"1.12.1" = {
x86_64-darwin-cpu = {
name = "libtorch-macos-1.12.1.zip";
url = "https://download.pytorch.org/libtorch/cpu/libtorch-macos-1.12.1.zip";
hash = "sha256-HSisxHs466c6XwvZEbkV/1kVNBzJOy3uVw9Bh497Vk8=";
};
x86_64-linux-cpu = {
name = "libtorch-cxx11-abi-shared-with-deps-1.12.1-cpu.zip";
url = "https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.12.1%2Bcpu.zip";
hash = "sha256-bHhC0WTli9vDJv56TaT6iA/d8im9zRkK1TlBuqkh4Wg=";
};
x86_64-linux-cuda = {
name = "libtorch-cxx11-abi-shared-with-deps-1.12.1-cu116.zip";
url = "https://download.pytorch.org/libtorch/cu116/libtorch-cxx11-abi-shared-with-deps-1.12.1%2Bcu116.zip";
hash = "sha256-YRcusDhrHYwIFOzt7vOuUlc11VyEUjIcBzjWEyi/874=";
};
};
} }