聂如 commited on
Commit
ac1764e
·
1 Parent(s): 7f58b70

Add design file

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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