alioth/nixpkgs
3
0
Fork 0
forked from mirrors/nixpkgs
Code Releases Activity

Merge pull request #155024 from prusnak/pytorch-bin-darwin

Browse source 
python3Packages.pytorch-bin: enable on darwin (cpu build)
This commit is contained in:
Pavol Rusnak 2022-01-14 20:32:26 +01:00 committed by GitHub
parent bc9fd41f23 ffde221dcb
commit 73757db5fb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -11,6 +11,7 @@
, patchelf
, pyyaml
, requests
, setuptools
, typing-extensions
}:
@ -41,6 +42,7 @@ in buildPythonPackage {
numpy
pyyaml
requests
setuptools
typing-extensions
];
@ -72,7 +74,7 @@ in buildPythonPackage {
# 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;
platforms = platforms.linux ++ platforms.darwin;
maintainers = with maintainers; [ junjihashimoto ];
};
}

25
pkgs/development/python-modules/pytorch/binary-hashes.nix
View file

@ -22,5 +22,30 @@ version : builtins.getAttr version {
url = "https://download.pytorch.org/whl/cu113/torch-1.10.0%2Bcu113-cp39-cp39-linux_x86_64.whl";
hash = "sha256-w8UJDh4b5cgDu7ZSvDoKzNH4hiXEyRfvpycNOg+wJOg=";
};
x86_64-darwin-37 = {
name = "torch-1.10.0-cp37-none-macosx_10_9_x86_64.whl";
url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp37-none-macosx_10_9_x86_64.whl";
hash = "sha256-RJkFVUcIfX736KdU8JwsTxRwKXrj5UkDY9umbHVQGyE=";
};
x86_64-darwin-38 = {
name = "torch-1.10.0-cp38-none-macosx_10_9_x86_64.whl";
url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp38-none-macosx_10_9_x86_64.whl";
hash = "sha256-rvevti6bF0tODl4eSkLjurO4SQpmjWZvYvfUUXVZ+/I=";
};
x86_64-darwin-39 = {
name = "torch-1.10.0-cp39-none-macosx_10_9_x86_64.whl";
url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp39-none-macosx_10_9_x86_64.whl";
hash = "sha256-1u+HRwtE35lw6EVCVH1bp3ILuJYWYCRB31VaObEk4rw=";
};
aarch64-darwin-38 = {
name = "torch-1.10.0-cp38-none-macosx_11_0_arm64.whl";
url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp38-none-macosx_11_0_arm64.whl";
hash = "sha256-1hhYJ7KFeAZTzdgdd6Cf3KdqWxkNWYbVUr4qXEQs+qQ=";
};
aarch64-darwin-39 = {
name = "torch-1.10.0-cp39-none-macosx_11_0_arm64.whl";
url = "https://download.pytorch.org/whl/cpu/torch-1.10.0-cp39-none-macosx_11_0_arm64.whl";
hash = "sha256-7qZ17AHsS0oGVf0phPFmpco7kz2uatTrTlLrpwJtwXY=";
};
};
}