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, )