mirror/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-09-29 15:43:00 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge pull request #77685 from bcdarwin/python3-ignite

Browse Source 
python3Packages.ignite: init at 0.2.1
This commit is contained in:
Mario Rodas 2020-02-27 07:25:22 -05:00 committed by GitHub
commit dfad6d1bd2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 71 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
pkgs/development/python-modules/ignite/default.nix Normal file
View File

@ -0,0 +1,38 @@
{ lib
, buildPythonPackage
, fetchFromGitHub
, pytest
, mock
, pytorch
, pynvml
, scikitlearn
, tqdm
}:
buildPythonPackage rec {
pname = "ignite";
version = "0.2.1";
src = fetchFromGitHub {
owner = "pytorch";
repo = pname;
rev = "v${version}";
sha256 = "15k6dd11yxn4923llcpmw4srl1by5ljhh7aw5pnkn4n4qpywh6cm";
};
checkInputs = [ pytest mock ];
checkPhase = ''
pytest -k 'not visdom and not tensorboard and not mlflow and not polyaxon' tests/
'';
# these packages are not currently in nixpkgs
propagatedBuildInputs = [ pytorch scikitlearn tqdm pynvml ];
meta = with lib; {
description = "High-level training library for PyTorch";
homepage = https://pytorch.org/ignite;
license = licenses.bsd3;
maintainers = [ maintainers.bcdarwin ];
};
}

29
pkgs/development/python-modules/pynvml/default.nix Normal file
View File

@ -0,0 +1,29 @@
{ lib
, buildPythonPackage
, fetchPypi
, pythonOlder
, cudatoolkit
}:
buildPythonPackage rec {
pname = "pynvml";
version = "8.0.4";
disabled = pythonOlder "3.6";
src = fetchPypi {
inherit pname version;
sha256 = "0pfykj1amqh1rixp90rg85v1nj6qmx89fahqr6ii4zlcckffmm68";
};
propagatedBuildInputs = [ cudatoolkit ];
doCheck = false; # no tests in PyPi dist
pythonImportsCheck = [ "pynvml" "pynvml.smi" ];
meta = with lib; {
description = "Python bindings for the NVIDIA Management Library";
homepage = https://www.nvidia.com;
license = licenses.bsd3;
maintainers = [ maintainers.bcdarwin ];
};
}

4
pkgs/top-level/python-packages.nix
View File

@ -783,6 +783,8 @@ in {
i3ipc = callPackage ../development/python-modules/i3ipc { };
ignite = callPackage ../development/python-modules/ignite { };
ihatemoney = callPackage ../development/python-modules/ihatemoney { };
imutils = callPackage ../development/python-modules/imutils { };
@ -5124,6 +5126,8 @@ in {
pynrrd = callPackage ../development/python-modules/pynrrd { };
pynvml = callPackage ../development/python-modules/pynvml { };
pynzb = callPackage ../development/python-modules/pynzb { };
process-tests = callPackage ../development/python-modules/process-tests { };