From fa1bc15e642c2a535b49deb856496cd01bedbe83 Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Sat, 28 Mar 2026 03:14:44 +0000 Subject: [PATCH] fix(installer): remove duplicate pip prefix from CUDA torch command --- installer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/installer.py b/installer.py index a701cb644..b07e75db8 100644 --- a/installer.py +++ b/installer.py @@ -559,8 +559,7 @@ def install_cuda(): if args.use_nightly: cmd = os.environ.get('TORCH_COMMAND', '--upgrade --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu128 --extra-index-url https://download.pytorch.org/whl/nightly/cu130') else: - # cmd = os.environ.get('TORCH_COMMAND', 'torch==2.10.0+cu128 torchvision==0.25.0+cu128 --index-url https://download.pytorch.org/whl/cu128') - cmd = os.environ.get("TORCH_COMMAND", "pip install -U torch==2.11.0+cu130 torchvision==0.26.0+cu130 --index-url https://download.pytorch.org/whl/cu130") + cmd = os.environ.get('TORCH_COMMAND', 'torch==2.11.0+cu130 torchvision==0.26.0+cu130 --index-url https://download.pytorch.org/whl/cu130') return cmd