Spaces:

zerchen
/

HORT

Running on Zero

zerchen commited on Apr 3

Commit

8e926b3

1 Parent(s): 53ccf28

fix a bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def install_cuda_toolkit():
     os.system(f"wget -q {CUDA_TOOLKIT_URL} -O {CUDA_TOOLKIT_FILE}")
     os.system(f"chmod +x {CUDA_TOOLKIT_FILE}")
     print("start to install cuda toolkit")
-    os.system("{CUDA_TOOLKIT_FILE} --silent --toolkit")
     os.environ["CUDA_HOME"] = "/usr/local/cuda"

     os.system(f"wget -q {CUDA_TOOLKIT_URL} -O {CUDA_TOOLKIT_FILE}")
     os.system(f"chmod +x {CUDA_TOOLKIT_FILE}")
     print("start to install cuda toolkit")
+    os.system(f"{CUDA_TOOLKIT_FILE} --silent --toolkit")
     os.environ["CUDA_HOME"] = "/usr/local/cuda"