movefasta/nixpkgs
0
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-07-11 21:06:19 +03:00
Code Activity

python3Packages.pytorch-bin: 1.9.1 -> 1.10.0

Browse source
This commit is contained in:
Junji Hashimoto 2021-11-10 00:25:41 +00:00
parent 14c30c8521
commit 207b16af90
3 changed files with 25 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

10
pkgs/development/python-modules/pytorch/bin.nix
View file

@ -18,7 +18,7 @@ let
pyVerNoDot = builtins.replaceStrings [ "." ] [ "" ] python.pythonVersion;
srcs = import ./binary-hashes.nix version;
unsupported = throw "Unsupported system";
version = "1.9.1";
version = "1.10.0";
in buildPythonPackage {
inherit version;
@ -59,13 +59,19 @@ in buildPythonPackage {
done
'';
# The wheel-binary is not stripped to avoid the error of `ImportError: libtorch_cuda_cpp.so: ELF load command address/offset not properly aligned.`.
dontStrip = true;
pythonImportsCheck = [ "torch" ];
meta = with lib; {
description = "Open source, prototype-to-production deep learning platform";
homepage = "https://pytorch.org/";
changelog = "https://github.com/pytorch/pytorch/releases/tag/v${version}";
license = licenses.unfree; # Includes CUDA and Intel MKL.
# Includes CUDA and Intel MKL, but redistributions of the binary are not limited.
# https://docs.nvidia.com/cuda/eula/index.html
# https://www.intel.com/content/www/us/en/developer/articles/license/onemkl-license-faq.html
license = licenses.bsd3;
platforms = platforms.linux;
maintainers = with maintainers; [ junjihashimoto ];
};