Spaces:
Running
on
L40S
Running
on
L40S
Upload app_hg.py with huggingface_hub
Browse files
app_hg.py
CHANGED
@@ -68,7 +68,7 @@ def install_cuda_toolkit():
|
|
68 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
69 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
70 |
|
71 |
-
subprocess.call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"]
|
72 |
import pytorch3d
|
73 |
|
74 |
install_cuda_toolkit()
|
|
|
68 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
69 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
70 |
|
71 |
+
subprocess.call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
|
72 |
import pytorch3d
|
73 |
|
74 |
install_cuda_toolkit()
|