diff options
author | Eric Sagnes <eric.sagnes@gmail.com> | 2017-08-19 16:47:23 +0900 |
---|---|---|
committer | Eric Sagnes <eric.sagnes@gmail.com> | 2017-08-21 20:43:28 +0900 |
commit | 14ce4217e31bb2c3e1dddc198b3b8110d868c690 (patch) | |
tree | 0784b342e6eb3958b892e0c36b527486dff893ac /pkgs/development/python-modules/torchvision | |
parent | dd809b930e4e1d5de7d1dd33310a8c28d9515992 (diff) | |
download | nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.tar nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.tar.gz nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.tar.bz2 nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.tar.lz nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.tar.xz nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.tar.zst nixpkgs-14ce4217e31bb2c3e1dddc198b3b8110d868c690.zip |
torchvision: init at 0.1.9
Diffstat (limited to 'pkgs/development/python-modules/torchvision')
-rw-r--r-- | pkgs/development/python-modules/torchvision/default.nix | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/torchvision/default.nix b/pkgs/development/python-modules/torchvision/default.nix new file mode 100644 index 00000000000..ebda3a92fc4 --- /dev/null +++ b/pkgs/development/python-modules/torchvision/default.nix @@ -0,0 +1,30 @@ +{ buildPythonPackage +, fetchPypi +, six +, numpy +, pillow +, pytorch +, lib }: + +buildPythonPackage rec { + version = "0.1.9"; + pname = "torchvision"; + name = "${pname}-${version}"; + + format = "wheel"; + + src = fetchPypi { + inherit pname version; + format = "wheel"; + sha256 = "016rjfh9w1x4xpw15ryxsvq3j2li17nd3a7qslnf3241hc6vdcwf"; + }; + + propagatedBuildInputs = [ six numpy pillow pytorch ]; + + meta = { + description = "PyTorch vision library"; + homepage = http://pytorch.org/; + license = lib.licenses.bsd3; + maintainers = with lib.maintainers; [ ericsagnes ]; + }; +} |