Spaces:

LTT
/

DiMeR

Runtime error

LutaoJiang commited on 25 days ago

Commit

548d313

1 Parent(s): 5be3795

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ def install_cuda_toolkit():
         "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
     )
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
-    os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
     print("==> finished installation")
 # Only execute CUDA installation in non-UI debug mode

         "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
     )
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
+    os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     print("==> finished installation")
 # Only execute CUDA installation in non-UI debug mode