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)
|