Spaces:

alexnasa
/

PartCrafter

Running on Zero

alexnasa commited on Jul 14

Commit

4f3b1e2

verified ·

1 Parent(s): 7ba8809

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,8 +44,10 @@ def install_cuda_toolkit():
     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     print("==> finished installation")
 install_cuda_toolkit()
 print(os.environ["CUDA_HOME"])

     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
     os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     print("==> finished installation")
+print("installing cuda toolkit")
 install_cuda_toolkit()
+peinr("finished")
 print(os.environ["CUDA_HOME"])