Use CUDA 12.4

Comfy-Org · Jan 27, 2025 · e281747 · e281747
1 parent 98bfef7
commit e281747
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comfy_cli/uv.py b/comfy_cli/uv.py
@@ -65,7 +65,7 @@ def parse_req_file(rf: PathLike, skips: Optional[list[str]] = None):
 
 class DependencyCompiler:
     rocmPytorchUrl = "https://download.pytorch.org/whl/rocm6.1"
-    nvidiaPytorchUrl = "https://download.pytorch.org/whl/cu121"
+    nvidiaPytorchUrl = "https://download.pytorch.org/whl/cu124"
 
     overrideGpu = dedent(
         """