tjh.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
14 format = "wheel";
15
16 src = fetchPypi {
17 inherit pname version;
18 format = "wheel";
19 sha256 = "18gvdabkmzfjg47ns0lw38mf85ry28nq1mas5rzlwvb4l5zmw2ms";
20 };
21
22 propagatedBuildInputs = [ six numpy pillow pytorch ];
23
24 meta = {
25 description = "PyTorch vision library";
26 homepage = "https://pytorch.org/";
27 license = lib.licenses.bsd3;
28 maintainers = with lib.maintainers; [ ericsagnes ];
29 };
30}