聂如
commited on
Commit
·
ac1764e
1
Parent(s):
7f58b70
Add design file
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -20,7 +20,7 @@ def install_cuda_toolkit():
|
|
20 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
21 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
22 |
|
23 |
-
install_cuda_toolkit() # to compile the dependencies
|
24 |
# # pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
25 |
# version_str="".join([
|
26 |
# f"py3{sys.version_info.minor}_cu",
|
|
|
20 |
# Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
|
21 |
os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
|
22 |
|
23 |
+
# install_cuda_toolkit() # to compile the dependencies
|
24 |
# # pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
|
25 |
# version_str="".join([
|
26 |
# f"py3{sys.version_info.minor}_cu",
|
requirements.txt
CHANGED
@@ -25,4 +25,5 @@ pathlib
|
|
25 |
nerfvis
|
26 |
h5py
|
27 |
joblib
|
28 |
-
numpy<2
|
|
|
|
25 |
nerfvis
|
26 |
h5py
|
27 |
joblib
|
28 |
+
numpy<2
|
29 |
+
git+https://github.com/facebookresearch/pytorch3d.git@stable
|