zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-torchaudio: add support for CUDA 12.5+ (#46746)

Browse Source
This commit is contained in:
Adam J. Stewart 2024-10-03 13:16:40 +02:00 committed by GitHub
parent 1c065f2da9
commit feabcb884a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
var/spack/repos/builtin/packages/py-torchaudio/package.py
View File

@ -106,7 +106,12 @@ class PyTorchaudio(PythonPackage):
depends_on("sox") depends_on("sox")
# https://github.com/pytorch/audio/pull/3811 # https://github.com/pytorch/audio/pull/3811
depends_on("cuda@:12.4", when="^py-torch+cuda") patch(
"https://github.com/pytorch/audio/pull/3811.patch?full_index=1",
sha256="34dce3403abb03f62827e8a1efcdb2bf7742477a01f155ebb9c7fefe9588b132",
when="@2.2:",
)
conflicts("^cuda@12.5:", when="@:2.1")
def setup_build_environment(self, env): def setup_build_environment(self, env):
# tools/setup_helpers/extension.py # tools/setup_helpers/extension.py