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