聂如 commited on
Commit
210470f
·
1 Parent(s): 2058342

Add design file

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -18,12 +18,12 @@ def install_cuda_toolkit():
18
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
19
 
20
  install_cuda_toolkit() # to compile the dependencies
21
- pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
22
- version_str="".join([
23
- f"py3{sys.version_info.minor}_cu",
24
- torch.version.cuda.replace(".",""),
25
- f"_pyt{pyt_version_str}"
26
- ])
27
  # install pytorch3d with the right version
28
  os.system('pip install iopath')
29
  os.system('FORCE_CUDA=1 pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"')
 
18
  os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
19
 
20
  install_cuda_toolkit() # to compile the dependencies
21
+ # # pyt_version_str=torch.__version__.split("+")[0].replace(".", "")
22
+ # version_str="".join([
23
+ # f"py3{sys.version_info.minor}_cu",
24
+ # torch.version.cuda.replace(".",""),
25
+ # f"_pyt{pyt_version_str}"
26
+ # ])
27
  # install pytorch3d with the right version
28
  os.system('pip install iopath')
29
  os.system('FORCE_CUDA=1 pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"')