diff --git a/pyproject.toml b/pyproject.toml
index 67b5950..d8b0c5c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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 = "*"