Merge pull request #232412 from GaetanLepage/torch-update

python3Packages.torch{audio, vision}: update
This commit is contained in:
Samuel Ainsworth 2023-05-17 15:48:24 -04:00 committed by GitHub
commit 699ab79713
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 6 deletions

View File

@ -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) [

View File

@ -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 = ''

View File

@ -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 ];