Add AMD support ot pyproject.toml

This commit is contained in:
VSlobolinskyi 2025-03-18 17:10:55 +02:00
parent 2ecc67f796
commit 0cbff69edf

View File

@ -8,17 +8,23 @@ package-mode = false
[tool.poetry.dependencies]
python = ">=3.11,<3.12"
# Platform-specific installations for torch and torchaudio:
# GPU-specific installations for torch and torchaudio:
# If AMD_GPU environment variable is set to "true", then use AMD-compatible packages.
torch = [
{ url = "https://download.pytorch.org/whl/cu118/torch-2.1.1%2Bcu118-cp311-cp311-win_amd64.whl#sha256=d99be44487d3ed0f7e6ef5d6689a37fb4a2f2821a9e7b59e7e04002a876a667a", markers = "sys_platform == 'win32'" },
{ url = "https://download.pytorch.org/whl/cu118/torch-2.1.1%2Bcu118-cp311-cp311-linux_x86_64.whl#sha256=f3c0ba02b50d0021ff26f030e22d4c45965537cf91f322e52a65b8c58396f81c", markers = "sys_platform == 'linux'" }
{ url = "https://download.pytorch.org/whl/cu118/torch-2.1.1%2Bcu118-cp311-cp311-win_amd64.whl#sha256=d99be44487d3ed0f7e6ef5d6689a37fb4a2f2821a9e7b59e7e04002a876a667a", markers = "sys_platform == 'win32' and platform_machine != 'AMD64'" },
{ url = "https://download.pytorch.org/whl/cu118/torch-2.1.1%2Bcu118-cp311-cp311-linux_x86_64.whl#sha256=f3c0ba02b50d0021ff26f030e22d4c45965537cf91f322e52a65b8c58396f81c", markers = "sys_platform == 'linux' and platform_machine != 'AMD64'" },
{ version = "^2.4.1", markers = "platform_machine == 'AMD64'" }
]
torchaudio = [
{ url = "https://download.pytorch.org/whl/cu118/torchaudio-2.1.1%2Bcu118-cp311-cp311-win_amd64.whl#sha256=79b5afa556063be18de4a1964339242301fe04e782e1030a22695257dd9afbd2", markers = "sys_platform == 'win32'" },
{ url = "https://download.pytorch.org/whl/cu118/torchaudio-2.1.1%2Bcu118-cp311-cp311-linux_x86_64.whl#sha256=2b077639f240176bb27e964e2e9b3a5c2a8d560a3a7bc1ffd0a024e81f2e10b4", markers = "sys_platform == 'linux'" }
{ url = "https://download.pytorch.org/whl/cu118/torchaudio-2.1.1%2Bcu118-cp311-cp311-win_amd64.whl#sha256=79b5afa556063be18de4a1964339242301fe04e782e1030a22695257dd9afbd2", markers = "sys_platform == 'win32' and platform_machine != 'AMD64'" },
{ url = "https://download.pytorch.org/whl/cu118/torchaudio-2.1.1%2Bcu118-cp311-cp311-linux_x86_64.whl#sha256=2b077639f240176bb27e964e2e9b3a5c2a8d560a3a7bc1ffd0a024e81f2e10b4", markers = "sys_platform == 'linux' and platform_machine != 'AMD64'" },
{ version = "2.4.1", markers = "platform_machine == 'AMD64'" }
]
# Additional dependency for AMD GPUs on Windows:
torch-directml = { version = "^0.2.5.dev240914", markers = "platform_machine == 'AMD64'" }
joblib = ">=1.1.0"
numba = "*"
numpy = "*"