聂如
commited on
Commit
·
210470f
1
Parent(s):
2058342
Add design file
Browse files
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 |
-
|
24 |
-
|
25 |
-
|
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"')
|