diff options
author | Alyssa Ross <hi@alyssa.is> | 2021-01-15 10:30:44 +0000 |
---|---|---|
committer | Alyssa Ross <hi@alyssa.is> | 2021-01-15 10:30:44 +0000 |
commit | e0794be8a0d11e90461e5a9c85012a36b93ec976 (patch) | |
tree | efd9cbc55ea3322867bf601c4d536758a3dd5fcc /nixpkgs/pkgs/development/python-modules/torchvision | |
parent | 3538874082ded7647b1ccec0343c7c1e882cfef3 (diff) | |
parent | 1a57d96edd156958b12782e8c8b6a374142a7248 (diff) | |
download | nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.tar nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.tar.gz nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.tar.bz2 nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.tar.lz nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.tar.xz nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.tar.zst nixlib-e0794be8a0d11e90461e5a9c85012a36b93ec976.zip |
Merge commit '1a57d96edd156958b12782e8c8b6a374142a7248'
Diffstat (limited to 'nixpkgs/pkgs/development/python-modules/torchvision')
-rw-r--r-- | nixpkgs/pkgs/development/python-modules/torchvision/default.nix | 54 |
1 files changed, 37 insertions, 17 deletions
diff --git a/nixpkgs/pkgs/development/python-modules/torchvision/default.nix b/nixpkgs/pkgs/development/python-modules/torchvision/default.nix index 4d3715790996..ea702a78fe1d 100644 --- a/nixpkgs/pkgs/development/python-modules/torchvision/default.nix +++ b/nixpkgs/pkgs/development/python-modules/torchvision/default.nix @@ -1,30 +1,50 @@ -{ buildPythonPackage -, fetchPypi -, six +{ stdenv +, buildPythonPackage +, fetchFromGitHub +, ninja +, which +, libjpeg_turbo +, libpng , numpy +, scipy , pillow , pytorch -, lib +, pytest }: buildPythonPackage rec { - version = "0.2.1"; - pname = "torchvision"; + pname = "torchvision"; + version = "0.8.2"; - format = "wheel"; - - src = fetchPypi { - inherit pname version; - format = "wheel"; - sha256 = "18gvdabkmzfjg47ns0lw38mf85ry28nq1mas5rzlwvb4l5zmw2ms"; + src = fetchFromGitHub { + owner = "pytorch"; + repo = "vision"; + rev = "v${version}"; + sha256 = "0yhpbq7linrk2qp5jxsvlgkmwa5bn38s9kcswy4jzvmx1fjbkpq0"; }; - propagatedBuildInputs = [ six numpy pillow pytorch ]; + nativeBuildInputs = [ libpng ninja which ]; + + TORCHVISION_INCLUDE = "${libjpeg_turbo.dev}/include/"; + TORCHVISION_LIBRARY = "${libjpeg_turbo}/lib/"; + + buildInputs = [ libjpeg_turbo libpng ]; + + propagatedBuildInputs = [ numpy pillow pytorch scipy ]; + + # tries to download many datasets for tests + doCheck = false; + + checkPhase = '' + HOME=$TMPDIR py.test test --ignore=test/test_datasets_download.py + ''; + + checkInputs = [ pytest ]; - meta = { + meta = with stdenv.lib; { description = "PyTorch vision library"; - homepage = "https://pytorch.org/"; - license = lib.licenses.bsd3; - maintainers = with lib.maintainers; [ ericsagnes ]; + homepage = "https://pytorch.org/"; + license = licenses.bsd3; + maintainers = with maintainers; [ ericsagnes SuperSandro2000 ]; }; } |