Spaces:

zerchen
/

HORT

Running on Zero

zerchen commited on Apr 3

Commit

671c0b1

1 Parent(s): 8e926b3

update cuda version

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ STEEL_BLUE=(0.2745098, 0.5098039, 0.7058824)
 def install_cuda_toolkit():
-    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
     print("start to download cuda toolkit")
     os.system(f"wget -q {CUDA_TOOLKIT_URL} -O {CUDA_TOOLKIT_FILE}")

 def install_cuda_toolkit():
+    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run"
     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
     print("start to download cuda toolkit")
     os.system(f"wget -q {CUDA_TOOLKIT_URL} -O {CUDA_TOOLKIT_FILE}")