tjh.dev
1{ buildPythonPackage
2, fetchPypi
3, six
4, numpy
5, pillow
6, pytorch
7, lib }:
8
9buildPythonPackage rec {
10 version = "0.1.9";
11 pname = "torchvision";
12 name = "${pname}-${version}";
13
14 format = "wheel";
15
16 src = fetchPypi {
17 inherit pname version;
18 format = "wheel";
19 sha256 = "016rjfh9w1x4xpw15ryxsvq3j2li17nd3a7qslnf3241hc6vdcwf";
20 };
21
22 propagatedBuildInputs = [ six numpy pillow pytorch ];
23
24 meta = {
25 description = "PyTorch vision library";
26 homepage = http://pytorch.org/;
27 license = lib.licenses.bsd3;
28 maintainers = with lib.maintainers; [ ericsagnes ];
29 };
30}