From 79670ab4a9fb685f2d8c80b87d7e7ecf78b6c969 Mon Sep 17 00:00:00 2001 From: LeoBorcherding Date: Wed, 6 May 2026 18:48:57 -0500 Subject: [PATCH] fix: use --no-deps for AMD Windows torch wheel install uv's resolver looks up rocm[libraries]==0.1.dev0 on PyPI during dependency resolution before downloading any wheels, and fails because the package doesn't exist on PyPI. --no-deps skips resolution entirely and installs all 5 AMD wheels directly. The GPU runtime dependency is satisfied by the HIP SDK, not a Python package. --- install.ps1 | 2 +- studio/install_python_stack.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/install.ps1 b/install.ps1 index 2563ddffae..9b55ff1492 100644 --- a/install.ps1 +++ b/install.ps1 @@ -1445,7 +1445,7 @@ shell.Run cmd, 0, False $rw0 = $ROCmAllWheelUrls[0]; $rw1 = $ROCmAllWheelUrls[1] $rw2 = $ROCmAllWheelUrls[2]; $rw3 = $ROCmAllWheelUrls[3] $rw4 = $ROCmAllWheelUrls[4] - $torchInstallExit = Invoke-InstallCommand { uv pip install --python $VenvPython --force-reinstall --no-cache-dir $rw0 $rw1 $rw2 $rw3 $rw4 } + $torchInstallExit = Invoke-InstallCommand { uv pip install --python $VenvPython --force-reinstall --no-cache-dir --no-deps $rw0 $rw1 $rw2 $rw3 $rw4 } if ($torchInstallExit -ne 0) { Write-Host "[ERROR] Failed to install AMD ROCm PyTorch (exit code $torchInstallExit)" -ForegroundColor Red return (Exit-InstallFailure "Failed to install AMD ROCm PyTorch (exit code $torchInstallExit)" $torchInstallExit) diff --git a/studio/install_python_stack.py b/studio/install_python_stack.py index 8ffd9a1974..a339d273bf 100644 --- a/studio/install_python_stack.py +++ b/studio/install_python_stack.py @@ -334,6 +334,7 @@ def _ensure_rocm_torch() -> None: f"ROCm torch (Windows, {rel_tag})", "--force-reinstall", "--no-cache-dir", + "--no-deps", *wheel_urls, constrain = False, )