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 |
try:
|
71 |
-
subprocess.check_call([sys.executable, "-m", "pip", "install", "git+https://github.com/facebookresearch/pytorch3d@stable"])
|
72 |
import pytorch3d
|
73 |
except Exception as err:
|
74 |
print(err)
|
|
|
68 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
69 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
70 |
try:
|
71 |
+
subprocess.check_call([sys.executable, "-m", "pip", "install", "-U", "git+https://github.com/facebookresearch/pytorch3d@stable"])
|
72 |
import pytorch3d
|
73 |
except Exception as err:
|
74 |
print(err)
|