Spaces:

stepfun-ai
/

Step1X-3D

Running on Zero

ReubenSun commited on 7 days ago

Commit

916dab8

1 Parent(s): 77948f7

cuda

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,6 +14,7 @@ def install_cuda_toolkit():
         os.environ["CUDA_HOME"],
         "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
     )
 install_cuda_toolkit()
 os.system('bash install.sh')
 import time

         os.environ["CUDA_HOME"],
         "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
     )
+    os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 install_cuda_toolkit()
 os.system('bash install.sh')
 import time