pythonPackages.pytorch: fix weirdly broken wheel version number
Otherwise, the wheel gets built with invalid metadata -- causing 'torch >= 1.0.0' to be unsatisfiable in other python packages, for instance. Signed-off-by: Austin Seipp <aseipp@pobox.com>
This commit is contained in:
parent
be96560327
commit
c6ecb3887f
1 changed files with 6 additions and 0 deletions
|
@ -58,6 +58,12 @@ in buildPythonPackage rec {
|
|||
done
|
||||
'';
|
||||
|
||||
# Override the (weirdly) wrong version set by default. See
|
||||
# https://github.com/NixOS/nixpkgs/pull/52437#issuecomment-449718038
|
||||
# https://github.com/pytorch/pytorch/blob/v1.0.0/setup.py#L267
|
||||
PYTORCH_BUILD_VERSION = version;
|
||||
PYTORCH_BUILD_NUMBER = 0;
|
||||
|
||||
buildInputs = [
|
||||
cmake
|
||||
numpy.blas
|
||||
|
|
Loading…
Reference in a new issue