revert stable cuda 13.2 to 13.0 (ampere segfault)

This commit is contained in:
CalamitousFelicitousness
2026-05-18 03:24:31 +01:00
parent 7ab3007e92
commit 10b652cc6c
+1 -1
View File
@@ -577,7 +577,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/cu132 --extra-index-url https://download.pytorch.org/whl/nightly/cu130')
else:
cmd = os.environ.get('TORCH_COMMAND', 'torch==2.12.0+cu132 torchvision==0.27.0+cu132 --index-url https://download.pytorch.org/whl/cu132')
cmd = os.environ.get('TORCH_COMMAND', 'torch==2.12.0+cu130 torchvision==0.27.0+cu130 --index-url https://download.pytorch.org/whl/cu130')
return cmd