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

python3Packages.pytorch-bin: 1.7.0 -> 1.7.1

Browse Source 
This also adds Python 3.9 compatibility.
This commit is contained in:
Daniël de Kok 2020-12-10 20:11:37 +01:00
parent 0f41c841e8
commit b51d8fc442
2 changed files with 9 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pkgs/development/python-modules/pytorch/bin.nix
View File

@ -3,6 +3,7 @@
, fetchurl , fetchurl
, isPy37 , isPy37
, isPy38 , isPy38
, isPy39
, python , python
, nvidia_x11 , nvidia_x11
, addOpenGLRunpath , addOpenGLRunpath
@ -19,7 +20,7 @@ let
platform = if stdenv.isDarwin then "darwin" else "linux"; platform = if stdenv.isDarwin then "darwin" else "linux";
srcs = import ./binary-hashes.nix version; srcs = import ./binary-hashes.nix version;
unsupported = throw "Unsupported system"; unsupported = throw "Unsupported system";
version = "1.7.0"; version = "1.7.1";
in buildPythonPackage { in buildPythonPackage {
inherit version; inherit version;
@ -28,7 +29,7 @@ in buildPythonPackage {
format = "wheel"; format = "wheel";
disabled = !(isPy37 || isPy38); disabled = !(isPy37 || isPy38 || isPy39);
src = fetchurl srcs."${stdenv.system}-${pyVerNoDot}" or unsupported; src = fetchurl srcs."${stdenv.system}-${pyVerNoDot}" or unsupported;
@ -45,16 +46,6 @@ in buildPythonPackage {
typing-extensions typing-extensions
]; ];
# PyTorch are broken: the dataclasses wheel is required, but ships with
# Python >= 3.7. Our dataclasses derivation is incompatible with >= 3.7.
#
# https://github.com/pytorch/pytorch/issues/46930
#
# Should be removed with the next PyTorch version.
pipInstallFlags = [
"--no-deps"
];
postInstall = '' postInstall = ''
# ONNX conversion # ONNX conversion
rm -rf $out/bin rm -rf $out/bin

8
pkgs/development/python-modules/pytorch/binary-hashes.nix
View File

@ -1,10 +1,14 @@
version: { version: {
x86_64-linux-37 = { x86_64-linux-37 = {
url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp37-cp37m-linux_x86_64.whl"; url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp37-cp37m-linux_x86_64.whl";
sha256 = "0cvafwzd44d7i2d0y01lyk75lhxffcf6c94cx0wla0cnhcg9mqlf"; hash = "sha256-XXbCVaQUhMHUGp/1cLnJ82y4XflCiqFaWK4WrHz8LqY=";
}; };
x86_64-linux-38 = { x86_64-linux-38 = {
url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp38-cp38-linux_x86_64.whl"; url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp38-cp38-linux_x86_64.whl";
sha256 = "1vxswkb2gzp87sbb5ycvyvs8aw1szp3v2mwdn4mp47yarlnbx8r6"; hash = "sha256-3S/GiAyV6DaWDYbvu8f2PTKH8uGJPFHTH5bb/gLw1z4=";
};
x86_64-linux-39 = {
url = "https://download.pytorch.org/whl/cu102/torch-${version}-cp39-cp39-linux_x86_64.whl";
hash = "sha256-o3k9zOsSseIoEpDMoSd8XOht39W/BE9lQoWk1pBXrqc=";
}; };
} }