Unverified Commit 699ab797 authored by Samuel Ainsworth's avatar Samuel Ainsworth Committed by GitHub
Browse files

Merge pull request #232412 from GaetanLepage/torch-update 

python3Packages.torch{audio, vision}: update
parents 72aa4e97 6c9fe0be

pkgs/development/python-modules/torch/default.nix

+2 −2
Original line number Diff line number Diff line
@@ -133,7 +133,7 @@ let 
in buildPythonPackage rec {

  pname = "torch";

  # Don't forget to update torch-bin to the same version.

  version = "2.0.0";

  version = "2.0.1";

  format = "setuptools";



  disabled = pythonOlder "3.8.0";
@@ -149,7 +149,7 @@ in buildPythonPackage rec { 
    repo = "pytorch";

    rev = "refs/tags/v${version}";

    fetchSubmodules = true;

    hash = "sha256-cSw7+AYBUcZLz3UyK/+JWWjQxKwVBXcFvBq0XAcL3tE=";

    hash = "sha256-xUj77yKz3IQ3gd/G32pI4OhL3LoN1zS7eFg0/0nZp5I=";

  };



  patches = lib.optionals (stdenv.isDarwin && stdenv.isx86_64) [

pkgs/development/python-modules/torchaudio/default.nix

+2 −2
Original line number Diff line number Diff line
@@ -12,13 +12,13 @@ 


buildPythonPackage rec {

  pname = "torchaudio";

  version = "2.0.1";

  version = "2.0.2";



  src = fetchFromGitHub {

    owner = "pytorch";

    repo = "audio";

    rev = "v${version}";

    hash = "sha256-qrDWFY+6eVV9prUzUzb5yzyFYtEvaSyEW0zeKqAg2Vk=";

    hash = "sha256-9lB4gLXq0nXHT1+DNOlbJQqNndt2I6kVoNwhMO/2qlE=";

  };



  postPatch = ''

pkgs/development/python-modules/torchvision/default.nix

+2 −2
Original line number Diff line number Diff line
@@ -40,7 +40,7 @@ let 
  };



  pname = "torchvision";

  version = "0.15.1";

  version = "0.15.2";

in

buildPythonPackage {

  inherit pname version;
@@ -49,7 +49,7 @@ buildPythonPackage { 
    owner = "pytorch";

    repo = "vision";

    rev = "refs/tags/v${version}";

    hash = "sha256-CQS2IXb8YSLrrkn/7BsO4Me5Cv0eXgMAKXM4rGzr0Bw=";

    hash = "sha256-KNbOgd6PCINZqZ24c/Ev+ODux3ik5iUlzem9uUfQArM=";

  };



  nativeBuildInputs = [ libpng ninja which ] ++ lib.optionals cudaSupport [ cuda-native-redist ];