python3Packages.torchaudio-bin: 0.13.1 -> 2.0.1 (64fb4575) · Commits · nix / nixpkgs

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

+31 −17

Original line number	Diff line number	Diff line
		{ lib
		, stdenv
		, addOpenGLRunpath
		, autoPatchelfHook
		, buildPythonPackage
		, cudaPackages
		, fetchurl
		, isPy37
		, isPy38
		, isPy39
		, isPy310
		, ffmpeg_4
		, pythonAtLeast
		, pythonOlder
		, python
		, torch-bin
		, pythonOlder
		, pythonAtLeast
		}:

		buildPythonPackage rec {
		pname = "torchaudio";
		version = "0.13.1";
		version = "2.0.1";
		format = "wheel";

		src =
		let pyVerNoDot = lib.replaceStrings [ "." ] [ "" ] python.pythonVersion;
		unsupported = throw "Unsupported system";
		srcs = (import ./binary-hashes.nix version)."${stdenv.system}-${pyVerNoDot}" or unsupported;
		in fetchurl srcs;
		in
		fetchurl srcs;

		disabled = (pythonOlder "3.8") \|\| (pythonAtLeast "3.12");

		buildInputs = with cudaPackages; [
		# $out/${sitePackages}/torchaudio/lib/libtorchaudio*.so wants libcudart.so.11.0 but torch/lib only ships
		# libcudart.$hash.so.11.0
		cuda_cudart

		# $out/${sitePackages}/torchaudio/lib/libtorchaudio*.so wants libnvToolsExt.so.2 but torch/lib only ships
		# libnvToolsExt-$hash.so.1
		cuda_nvtx

		disabled = !(isPy38 \|\| isPy39 \|\| isPy310);
		ffmpeg_4.lib
		];

		nativeBuildInputs = [
		autoPatchelfHook
		addOpenGLRunpath
		];

		propagatedBuildInputs = [
		torch-bin
		];

		preInstall = ''
		addAutoPatchelfSearchPath "${torch-bin}/${python.sitePackages}/torch"
		'';

		# The wheel-binary is not stripped to avoid the error of `ImportError: libtorch_cuda_cpp.so: ELF load command address/offset not properly aligned.`.
		dontStrip = true;

		pythonImportsCheck = [ "torchaudio" ];

		postFixup = ''
		# Note: after patchelf'ing, libcudart can still not be found. However, this should
		# not be an issue, because PyTorch is loaded before torchvision and brings
		# in the necessary symbols.
		patchelf --set-rpath "${lib.makeLibraryPath [ stdenv.cc.cc.lib ]}:${torch-bin}/${python.sitePackages}/torch/lib:" \
		"$out/${python.sitePackages}/torchaudio/_torchaudio.so"
		'';

		meta = with lib; {
		description = "PyTorch audio library";
		homepage = "https://pytorch.org/";

pkgs/development/python-modules/torchaudio/binary-hashes.nix

+57 −37

Original line number	Diff line number	Diff line
		@@ -6,66 +6,86 @@
		# To add a new version, run "prefetch.sh 'new-version'" to paste the generated file as follows.

		version : builtins.getAttr version {
		"0.13.1" = {
		"2.0.1" = {
		x86_64-linux-38 = {
		name = "torchaudio-0.13.1-cp38-cp38-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu117/torchaudio-0.13.1%2Bcu117-cp38-cp38-linux_x86_64.whl";
		hash = "sha256-QCY7LUVyj7/x2zOBJyvkKXD/blj5KZSqWHKlvUx+cmQ=";
		name = "torchaudio-2.0.1-cp38-cp38-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu118/torchaudio-2.0.1%2Bcu118-cp38-cp38-linux_x86_64.whl";
		hash = "sha256-lLDpx2ypHR4CiYlZIPv+jBF0ZNdXtktd+tsTCM+ZBPk=";
		};
		x86_64-linux-39 = {
		name = "torchaudio-0.13.1-cp39-cp39-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu117/torchaudio-0.13.1%2Bcu117-cp39-cp39-linux_x86_64.whl";
		hash = "sha256-Zbs2FdQz1bkwrNwQNu+xJAR9VxfbpN63D0GSkNlC+DY=";
		name = "torchaudio-2.0.1-cp39-cp39-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu118/torchaudio-2.0.1%2Bcu118-cp39-cp39-linux_x86_64.whl";
		hash = "sha256-Bws4SWlhQr49keCycHbaHz+MtDKrzONc2VbRkfwNgYc=";
		};
		x86_64-linux-310 = {
		name = "torchaudio-0.13.1-cp310-cp310-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu117/torchaudio-0.13.1%2Bcu117-cp310-cp310-linux_x86_64.whl";
		hash = "sha256-k/RVSktT+WmNAiJJA8kjwSpsIrPJQtz8IXm1gdjzcUY=";
		name = "torchaudio-2.0.1-cp310-cp310-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu118/torchaudio-2.0.1%2Bcu118-cp310-cp310-linux_x86_64.whl";
		hash = "sha256-GcTvkBIyTE+4DqZpNFUbeAfZcUjChTji6rr+FqtQ6Rw=";
		};
		x86_64-linux-311 = {
		name = "torchaudio-2.0.1-cp311-cp311-linux_x86_64.whl";
		url = "https://download.pytorch.org/whl/cu118/torchaudio-2.0.1%2Bcu118-cp311-cp311-linux_x86_64.whl";
		hash = "sha256-GicitvZleO3FY+d7TMB6ItZjorte5cneJTlmGpihTbk=";
		};
		x86_64-darwin-38 = {
		name = "torchaudio-0.13.1-cp38-cp38-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-0.13.1-cp38-cp38-macosx_10_9_x86_64.whl";
		hash = "sha256-Qs5cZtMEvCzWgziRa4Ij4yLgmoTcvZIogU7za8R3o3s=";
		name = "torchaudio-2.0.1-cp38-cp38-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp38-cp38-macosx_10_9_x86_64.whl";
		hash = "sha256-AiyhuqS7gZt4NDvUe1f/bcb5/Bn6TvJplGqt9+Yts8A=";
		};
		x86_64-darwin-39 = {
		name = "torchaudio-0.13.1-cp39-cp39-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-0.13.1-cp39-cp39-macosx_10_9_x86_64.whl";
		hash = "sha256-nSFwVA3jKuAxqrOTYSmGjoluoEFhe21mkt3mqi37CiM=";
		name = "torchaudio-2.0.1-cp39-cp39-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp39-cp39-macosx_10_9_x86_64.whl";
		hash = "sha256-48bI+eqfDi33oLk3Ww3PlVkG44/BL6tUK3KoYVZK+Oc=";
		};
		x86_64-darwin-310 = {
		name = "torchaudio-0.13.1-cp310-cp310-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-0.13.1-cp310-cp310-macosx_10_9_x86_64.whl";
		hash = "sha256-Xg89xmmVBlITZCZnBOa/idDQV5/UNdEsXC9YWNUt5Po=";
		name = "torchaudio-2.0.1-cp310-cp310-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp310-cp310-macosx_10_9_x86_64.whl";
		hash = "sha256-tdIeu7VecEDUGNUGKw6IL5Zg1otHezj9Q2+mySzLtSo=";
		};
		x86_64-darwin-311 = {
		name = "torchaudio-2.0.1-cp311-cp311-macosx_10_9_x86_64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp311-cp311-macosx_10_9_x86_64.whl";
		hash = "sha256-4qBHZ1STwKolj+xiHvQOiwGr49jbyHIVLktZmEGKo8U=";
		};
		aarch64-darwin-38 = {
		name = "torchaudio-0.13.1-cp38-cp38-macosx_12_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-0.13.1-cp38-cp38-macosx_12_0_arm64.whl";
		hash = "sha256-sJOz52YchRaOyd3iz5c0WWXqCTHT0qfni9QJIh5taZg=";
		name = "torchaudio-2.0.1-cp38-cp38-macosx_11_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp38-cp38-macosx_11_0_arm64.whl";
		hash = "sha256-oVOtXNti3o7J/RNgoNCAu6851XiuBOeI2yEVceZ1t+A=";
		};
		aarch64-darwin-39 = {
		name = "torchaudio-0.13.1-cp39-cp39-macosx_12_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-0.13.1-cp39-cp39-macosx_12_0_arm64.whl";
		hash = "sha256-kfz79HAAQC0Sv/JiTmIgoP07jKjub/Ue31lF7DmrCn8=";
		name = "torchaudio-2.0.1-cp39-cp39-macosx_11_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp39-cp39-macosx_11_0_arm64.whl";
		hash = "sha256-HQzwd5ozTsGGHp+ii862amM8Quj2szIuLjf/nyDQroE=";
		};
		aarch64-darwin-310 = {
		name = "torchaudio-0.13.1-cp310-cp310-macosx_12_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-0.13.1-cp310-cp310-macosx_12_0_arm64.whl";
		hash = "sha256-7HKhfU0heIKed4BoKZm1Nc9X/hYNDCCw1r3BrRqHxN0=";
		name = "torchaudio-2.0.1-cp310-cp310-macosx_11_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp310-cp310-macosx_11_0_arm64.whl";
		hash = "sha256-bbzZOynXGi9QDzajTqXkZ/UQ93PahTIgmOa92MncmUg=";
		};
		aarch64-darwin-311 = {
		name = "torchaudio-2.0.1-cp311-cp311-macosx_11_0_arm64.whl";
		url = "https://download.pytorch.org/whl/cpu/torchaudio-2.0.1-cp311-cp311-macosx_11_0_arm64.whl";
		hash = "sha256-kaKOWH9wigMyDt28xKfdGtcVCz1IRrbBVX2FzImo0Gw=";
		};
		aarch64-linux-38 = {
		name = "torchaudio-0.13.1-cp38-cp38-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-0.13.1-cp38-cp38-manylinux2014_aarch64.whl";
		hash = "sha256-PEi8/wDq6BgPh/WNHJ5+n9jEy36z6ogXk1+2BI0VK8c=";
		name = "torchaudio-2.0.1-cp38-cp38-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-2.0.1-cp38-cp38-manylinux2014_aarch64.whl";
		hash = "sha256-qlsjzsMVcWRKpn1s1QGzl/QmVX3F8y6NtuzbJ7GUClg=";
		};
		aarch64-linux-39 = {
		name = "torchaudio-0.13.1-cp39-cp39-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-0.13.1-cp39-cp39-manylinux2014_aarch64.whl";
		hash = "sha256-MCOutcGRBHvvFoGjdBv/1KIWS1imTK0k3TfaXhrC0fE=";
		name = "torchaudio-2.0.1-cp39-cp39-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-2.0.1-cp39-cp39-manylinux2014_aarch64.whl";
		hash = "sha256-ry16SysebkDnnA7d0Qezu0MVkdBJTX+X7ffBhkN7XBo=";
		};
		aarch64-linux-310 = {
		name = "torchaudio-0.13.1-cp310-cp310-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-0.13.1-cp310-cp310-manylinux2014_aarch64.whl";
		hash = "sha256-LkdWLNzdR8uO2Go88FO3BnzJ6INA9FUK5z15DdvBLyE=";
		name = "torchaudio-2.0.1-cp310-cp310-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-2.0.1-cp310-cp310-manylinux2014_aarch64.whl";
		hash = "sha256-6f/6Y0Gbxl7Pg5Vo3QS+O6VibF9bJsmlZsA4KtKXcck=";
		};
		aarch64-linux-311 = {
		name = "torchaudio-2.0.1-cp311-cp311-manylinux2014_aarch64.whl";
		url = "https://download.pytorch.org/whl/torchaudio-2.0.1-cp311-cp311-manylinux2014_aarch64.whl";
		hash = "sha256-MkQuKxHfwJxMW2zEuSTT84wslGPuKOSGUi+fSLCbf7c=";
		};
		};
		}

pkgs/development/python-modules/torchaudio/prefetch.sh

+9 −5

Original line number	Diff line number	Diff line
		@@ -5,24 +5,28 @@ set -eou pipefail

		version=$1

		linux_cuda_version="cu117"
		linux_cuda_version="cu118"
		linux_cuda_bucket="https://download.pytorch.org/whl/${linux_cuda_version}"
		linux_cpu_bucket="https://download.pytorch.org/whl"
		darwin_bucket="https://download.pytorch.org/whl"
		darwin_bucket="https://download.pytorch.org/whl/cpu"

		url_and_key_list=(
		"x86_64-linux-38 $linux_cuda_bucket/torchaudio-${version}%2B${linux_cuda_version}-cp38-cp38-linux_x86_64.whl torchaudio-${version}-cp38-cp38-linux_x86_64.whl"
		"x86_64-linux-39 $linux_cuda_bucket/torchaudio-${version}%2B${linux_cuda_version}-cp39-cp39-linux_x86_64.whl torchaudio-${version}-cp39-cp39-linux_x86_64.whl"
		"x86_64-linux-310 $linux_cuda_bucket/torchaudio-${version}%2B${linux_cuda_version}-cp310-cp310-linux_x86_64.whl torchaudio-${version}-cp310-cp310-linux_x86_64.whl"
		"x86_64-linux-311 $linux_cuda_bucket/torchaudio-${version}%2B${linux_cuda_version}-cp311-cp311-linux_x86_64.whl torchaudio-${version}-cp311-cp311-linux_x86_64.whl"
		"x86_64-darwin-38 $darwin_bucket/torchaudio-${version}-cp38-cp38-macosx_10_9_x86_64.whl torchaudio-${version}-cp38-cp38-macosx_10_9_x86_64.whl"
		"x86_64-darwin-39 $darwin_bucket/torchaudio-${version}-cp39-cp39-macosx_10_9_x86_64.whl torchaudio-${version}-cp39-cp39-macosx_10_9_x86_64.whl"
		"x86_64-darwin-310 $darwin_bucket/torchaudio-${version}-cp310-cp310-macosx_10_9_x86_64.whl torchaudio-${version}-cp310-cp310-macosx_10_9_x86_64.whl"
		"aarch64-darwin-38 $darwin_bucket/cpu/torchaudio-${version}-cp38-cp38-macosx_12_0_arm64.whl torchaudio-${version}-cp38-cp38-macosx_12_0_arm64.whl"
		"aarch64-darwin-39 $darwin_bucket/cpu/torchaudio-${version}-cp39-cp39-macosx_12_0_arm64.whl torchaudio-${version}-cp39-cp39-macosx_12_0_arm64.whl"
		"aarch64-darwin-310 $darwin_bucket/cpu/torchaudio-${version}-cp310-cp310-macosx_12_0_arm64.whl torchaudio-${version}-cp310-cp310-macosx_12_0_arm64.whl"
		"x86_64-darwin-311 $darwin_bucket/torchaudio-${version}-cp311-cp311-macosx_10_9_x86_64.whl torchaudio-${version}-cp311-cp311-macosx_10_9_x86_64.whl"
		"aarch64-darwin-38 $darwin_bucket/torchaudio-${version}-cp38-cp38-macosx_11_0_arm64.whl torchaudio-${version}-cp38-cp38-macosx_11_0_arm64.whl"
		"aarch64-darwin-39 $darwin_bucket/torchaudio-${version}-cp39-cp39-macosx_11_0_arm64.whl torchaudio-${version}-cp39-cp39-macosx_11_0_arm64.whl"
		"aarch64-darwin-310 $darwin_bucket/torchaudio-${version}-cp310-cp310-macosx_11_0_arm64.whl torchaudio-${version}-cp310-cp310-macosx_11_0_arm64.whl"
		"aarch64-darwin-311 $darwin_bucket/torchaudio-${version}-cp311-cp311-macosx_11_0_arm64.whl torchaudio-${version}-cp311-cp311-macosx_11_0_arm64.whl"
		"aarch64-linux-38 $linux_cpu_bucket/torchaudio-${version}-cp38-cp38-manylinux2014_aarch64.whl torchaudio-${version}-cp38-cp38-manylinux2014_aarch64.whl"
		"aarch64-linux-39 $linux_cpu_bucket/torchaudio-${version}-cp39-cp39-manylinux2014_aarch64.whl torchaudio-${version}-cp39-cp39-manylinux2014_aarch64.whl"
		"aarch64-linux-310 $linux_cpu_bucket/torchaudio-${version}-cp310-cp310-manylinux2014_aarch64.whl torchaudio-${version}-cp310-cp310-manylinux2014_aarch64.whl"
		"aarch64-linux-311 $linux_cpu_bucket/torchaudio-${version}-cp311-cp311-manylinux2014_aarch64.whl torchaudio-${version}-cp311-cp311-manylinux2014_aarch64.whl"
		)

		hashfile=binary-hashes-"$version".nix

Admin message