Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-03 03:20:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(installer): remove duplicate pip prefix from CUDA torch command

Browse Source
This commit is contained in:
CalamitousFelicitousness
2026-03-28 03:14:44 +00:00
parent 6eadf40f33
commit fa1bc15e64
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
installer.py
+1 -2
View File
@@ -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