Skip to content

Commit 31e3488

Browse files
Roeia99facebook-github-bot
Roeia99
authored and
facebook-github-bot
committed
Changed tutorials' pip searching
Summary: ### Generalise tutorials' pip searching: ## Required Information: This diff contains changes to several PyTorch3D tutorials. **Purpose of this diff:** Replace the current installation code with a more streamlined approach that tries to install the wheel first and falls back to installing from source if the wheel is not found. **Why this diff is required:** This diff makes it easier to cope with new PyTorch releases and reduce the need for manual intervention, as the current process involves checking the version of PyTorch in Colab and building a new wheel if it doesn't match the expected version, which generates additional work each time there is a a new PyTorch version in Colab. **Changes:** Before: ``` if torch.__version__.startswith("2.1.") and sys.platform.startswith("linux"): # We try to install PyTorch3D via a released wheel. pyt_version_str=torch.__version__.split("+")[0].replace(".", "") version_str="".join([ f"py3{sys.version_info.minor}_cu", torch.version.cuda.replace(".",""), f"_pyt{pyt_version_str}" ]) !pip install fvcore iopath !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html else: # We try to install PyTorch3D from source. !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable' ``` After: ``` pyt_version_str=torch.__version__.split("+")[0].replace(".", "") version_str="".join([ f"py3{sys.version_info.minor}_cu", torch.version.cuda.replace(".",""), f"_pyt{pyt_version_str}" ]) !pip install fvcore iopath if sys.platform.startswith("linux"): # We try to install PyTorch3D via a released wheel. !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html pip_list = !pip freeze need_pytorch3d = not any(i.startswith("pytorch3d==") for i in pip_list) if need_pytorch3d: # We try to install PyTorch3D from source. !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable' ``` Reviewed By: bottler Differential Revision: D55431832 fbshipit-source-id: a8de9162470698320241ae8401427dcb1ce17c37
1 parent b215776 commit 31e3488

12 files changed

+216
-144
lines changed

‎docs/tutorials/bundle_adjustment.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -83,25 +83,31 @@
8383
"import os\n",
8484
"import sys\n",
8585
"import torch\n",
86+
"import subprocess\n",
8687
"need_pytorch3d=False\n",
8788
"try:\n",
8889
" import pytorch3d\n",
8990
"except ModuleNotFoundError:\n",
9091
" need_pytorch3d=True\n",
9192
"if need_pytorch3d:\n",
92-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
93-
" # We try to install PyTorch3D via a released wheel.\n",
94-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
95-
" version_str=\"\".join([\n",
96-
" f\"py3{sys.version_info.minor}_cu\",\n",
97-
" torch.version.cuda.replace(\".\",\"\"),\n",
98-
" f\"_pyt{pyt_version_str}\"\n",
99-
" ])\n",
100-
" !pip install fvcore iopath\n",
93+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
94+
" version_str=\"\".join([\n",
95+
" f\"py3{sys.version_info.minor}_cu\",\n",
96+
" torch.version.cuda.replace(\".\",\"\"),\n",
97+
" f\"_pyt{pyt_version_str}\"\n",
98+
" ])\n",
99+
" !pip install fvcore iopath\n",
100+
" if sys.platform.startswith(\"linux\"):\n",
101+
" print(\"Trying to install wheel for PyTorch3D\")\n",
101102
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
102-
" else:\n",
103-
" # We try to install PyTorch3D from source.\n",
104-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
103+
" pip_list = !pip freeze\n",
104+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
105+
" if need_pytorch3d:\n",
106+
" print(f\"failed to find/install wheel for {version_str}\")\n",
107+
"if need_pytorch3d:\n",
108+
" print(\"Installing PyTorch3D from source\")\n",
109+
" !pip install ninja\n",
110+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
105111
]
106112
},
107113
{

‎docs/tutorials/camera_position_optimization_with_differentiable_rendering.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -70,25 +70,31 @@
7070
"import os\n",
7171
"import sys\n",
7272
"import torch\n",
73+
"import subprocess\n",
7374
"need_pytorch3d=False\n",
7475
"try:\n",
7576
" import pytorch3d\n",
7677
"except ModuleNotFoundError:\n",
7778
" need_pytorch3d=True\n",
7879
"if need_pytorch3d:\n",
79-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
80-
" # We try to install PyTorch3D via a released wheel.\n",
81-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
82-
" version_str=\"\".join([\n",
83-
" f\"py3{sys.version_info.minor}_cu\",\n",
84-
" torch.version.cuda.replace(\".\",\"\"),\n",
85-
" f\"_pyt{pyt_version_str}\"\n",
86-
" ])\n",
87-
" !pip install fvcore iopath\n",
80+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
81+
" version_str=\"\".join([\n",
82+
" f\"py3{sys.version_info.minor}_cu\",\n",
83+
" torch.version.cuda.replace(\".\",\"\"),\n",
84+
" f\"_pyt{pyt_version_str}\"\n",
85+
" ])\n",
86+
" !pip install fvcore iopath\n",
87+
" if sys.platform.startswith(\"linux\"):\n",
88+
" print(\"Trying to install wheel for PyTorch3D\")\n",
8889
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
89-
" else:\n",
90-
" # We try to install PyTorch3D from source.\n",
91-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
90+
" pip_list = !pip freeze\n",
91+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
92+
" if need_pytorch3d:\n",
93+
" print(f\"failed to find/install wheel for {version_str}\")\n",
94+
"if need_pytorch3d:\n",
95+
" print(\"Installing PyTorch3D from source\")\n",
96+
" !pip install ninja\n",
97+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
9298
]
9399
},
94100
{

‎docs/tutorials/dataloaders_ShapeNetCore_R2N2.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -45,25 +45,31 @@
4545
"import os\n",
4646
"import sys\n",
4747
"import torch\n",
48+
"import subprocess\n",
4849
"need_pytorch3d=False\n",
4950
"try:\n",
5051
" import pytorch3d\n",
5152
"except ModuleNotFoundError:\n",
5253
" need_pytorch3d=True\n",
5354
"if need_pytorch3d:\n",
54-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
55-
" # We try to install PyTorch3D via a released wheel.\n",
56-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
57-
" version_str=\"\".join([\n",
58-
" f\"py3{sys.version_info.minor}_cu\",\n",
59-
" torch.version.cuda.replace(\".\",\"\"),\n",
60-
" f\"_pyt{pyt_version_str}\"\n",
61-
" ])\n",
62-
" !pip install fvcore iopath\n",
55+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
56+
" version_str=\"\".join([\n",
57+
" f\"py3{sys.version_info.minor}_cu\",\n",
58+
" torch.version.cuda.replace(\".\",\"\"),\n",
59+
" f\"_pyt{pyt_version_str}\"\n",
60+
" ])\n",
61+
" !pip install fvcore iopath\n",
62+
" if sys.platform.startswith(\"linux\"):\n",
63+
" print(\"Trying to install wheel for PyTorch3D\")\n",
6364
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
64-
" else:\n",
65-
" # We try to install PyTorch3D from source.\n",
66-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
65+
" pip_list = !pip freeze\n",
66+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
67+
" if need_pytorch3d:\n",
68+
" print(f\"failed to find/install wheel for {version_str}\")\n",
69+
"if need_pytorch3d:\n",
70+
" print(\"Installing PyTorch3D from source\")\n",
71+
" !pip install ninja\n",
72+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
6773
]
6874
},
6975
{

‎docs/tutorials/deform_source_mesh_to_target_mesh.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -84,25 +84,31 @@
8484
"import os\n",
8585
"import sys\n",
8686
"import torch\n",
87+
"import subprocess\n",
8788
"need_pytorch3d=False\n",
8889
"try:\n",
8990
" import pytorch3d\n",
9091
"except ModuleNotFoundError:\n",
9192
" need_pytorch3d=True\n",
9293
"if need_pytorch3d:\n",
93-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
94-
" # We try to install PyTorch3D via a released wheel.\n",
95-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
96-
" version_str=\"\".join([\n",
97-
" f\"py3{sys.version_info.minor}_cu\",\n",
98-
" torch.version.cuda.replace(\".\",\"\"),\n",
99-
" f\"_pyt{pyt_version_str}\"\n",
100-
" ])\n",
101-
" !pip install fvcore iopath\n",
94+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
95+
" version_str=\"\".join([\n",
96+
" f\"py3{sys.version_info.minor}_cu\",\n",
97+
" torch.version.cuda.replace(\".\",\"\"),\n",
98+
" f\"_pyt{pyt_version_str}\"\n",
99+
" ])\n",
100+
" !pip install fvcore iopath\n",
101+
" if sys.platform.startswith(\"linux\"):\n",
102+
" print(\"Trying to install wheel for PyTorch3D\")\n",
102103
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
103-
" else:\n",
104-
" # We try to install PyTorch3D from source.\n",
105-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
104+
" pip_list = !pip freeze\n",
105+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
106+
" if need_pytorch3d:\n",
107+
" print(f\"failed to find/install wheel for {version_str}\")\n",
108+
"if need_pytorch3d:\n",
109+
" print(\"Installing PyTorch3D from source\")\n",
110+
" !pip install ninja\n",
111+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
106112
]
107113
},
108114
{

‎docs/tutorials/fit_simple_neural_radiance_field.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -50,25 +50,31 @@
5050
"import os\n",
5151
"import sys\n",
5252
"import torch\n",
53+
"import subprocess\n",
5354
"need_pytorch3d=False\n",
5455
"try:\n",
5556
" import pytorch3d\n",
5657
"except ModuleNotFoundError:\n",
5758
" need_pytorch3d=True\n",
5859
"if need_pytorch3d:\n",
59-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
60-
" # We try to install PyTorch3D via a released wheel.\n",
61-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
62-
" version_str=\"\".join([\n",
63-
" f\"py3{sys.version_info.minor}_cu\",\n",
64-
" torch.version.cuda.replace(\".\",\"\"),\n",
65-
" f\"_pyt{pyt_version_str}\"\n",
66-
" ])\n",
67-
" !pip install fvcore iopath\n",
60+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
61+
" version_str=\"\".join([\n",
62+
" f\"py3{sys.version_info.minor}_cu\",\n",
63+
" torch.version.cuda.replace(\".\",\"\"),\n",
64+
" f\"_pyt{pyt_version_str}\"\n",
65+
" ])\n",
66+
" !pip install fvcore iopath\n",
67+
" if sys.platform.startswith(\"linux\"):\n",
68+
" print(\"Trying to install wheel for PyTorch3D\")\n",
6869
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
69-
" else:\n",
70-
" # We try to install PyTorch3D from source.\n",
71-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
70+
" pip_list = !pip freeze\n",
71+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
72+
" if need_pytorch3d:\n",
73+
" print(f\"failed to find/install wheel for {version_str}\")\n",
74+
"if need_pytorch3d:\n",
75+
" print(\"Installing PyTorch3D from source\")\n",
76+
" !pip install ninja\n",
77+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
7278
]
7379
},
7480
{

‎docs/tutorials/fit_textured_mesh.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -62,25 +62,31 @@
6262
"import os\n",
6363
"import sys\n",
6464
"import torch\n",
65+
"import subprocess\n",
6566
"need_pytorch3d=False\n",
6667
"try:\n",
6768
" import pytorch3d\n",
6869
"except ModuleNotFoundError:\n",
6970
" need_pytorch3d=True\n",
7071
"if need_pytorch3d:\n",
71-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
72-
" # We try to install PyTorch3D via a released wheel.\n",
73-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
74-
" version_str=\"\".join([\n",
75-
" f\"py3{sys.version_info.minor}_cu\",\n",
76-
" torch.version.cuda.replace(\".\",\"\"),\n",
77-
" f\"_pyt{pyt_version_str}\"\n",
78-
" ])\n",
79-
" !pip install fvcore iopath\n",
72+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
73+
" version_str=\"\".join([\n",
74+
" f\"py3{sys.version_info.minor}_cu\",\n",
75+
" torch.version.cuda.replace(\".\",\"\"),\n",
76+
" f\"_pyt{pyt_version_str}\"\n",
77+
" ])\n",
78+
" !pip install fvcore iopath\n",
79+
" if sys.platform.startswith(\"linux\"):\n",
80+
" print(\"Trying to install wheel for PyTorch3D\")\n",
8081
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
81-
" else:\n",
82-
" # We try to install PyTorch3D from source.\n",
83-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
82+
" pip_list = !pip freeze\n",
83+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
84+
" if need_pytorch3d:\n",
85+
" print(f\"failed to find/install wheel for {version_str}\")\n",
86+
"if need_pytorch3d:\n",
87+
" print(\"Installing PyTorch3D from source\")\n",
88+
" !pip install ninja\n",
89+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
8490
]
8591
},
8692
{

‎docs/tutorials/fit_textured_volume.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -41,25 +41,31 @@
4141
"import os\n",
4242
"import sys\n",
4343
"import torch\n",
44+
"import subprocess\n",
4445
"need_pytorch3d=False\n",
4546
"try:\n",
4647
" import pytorch3d\n",
4748
"except ModuleNotFoundError:\n",
4849
" need_pytorch3d=True\n",
4950
"if need_pytorch3d:\n",
50-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
51-
" # We try to install PyTorch3D via a released wheel.\n",
52-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
53-
" version_str=\"\".join([\n",
54-
" f\"py3{sys.version_info.minor}_cu\",\n",
55-
" torch.version.cuda.replace(\".\",\"\"),\n",
56-
" f\"_pyt{pyt_version_str}\"\n",
57-
" ])\n",
58-
" !pip install fvcore iopath\n",
51+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
52+
" version_str=\"\".join([\n",
53+
" f\"py3{sys.version_info.minor}_cu\",\n",
54+
" torch.version.cuda.replace(\".\",\"\"),\n",
55+
" f\"_pyt{pyt_version_str}\"\n",
56+
" ])\n",
57+
" !pip install fvcore iopath\n",
58+
" if sys.platform.startswith(\"linux\"):\n",
59+
" print(\"Trying to install wheel for PyTorch3D\")\n",
5960
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
60-
" else:\n",
61-
" # We try to install PyTorch3D from source.\n",
62-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
61+
" pip_list = !pip freeze\n",
62+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
63+
" if need_pytorch3d:\n",
64+
" print(f\"failed to find/install wheel for {version_str}\")\n",
65+
"if need_pytorch3d:\n",
66+
" print(\"Installing PyTorch3D from source\")\n",
67+
" !pip install ninja\n",
68+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
6369
]
6470
},
6571
{

‎docs/tutorials/implicitron_config_system.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -72,25 +72,31 @@
7272
"import os\n",
7373
"import sys\n",
7474
"import torch\n",
75+
"import subprocess\n",
7576
"need_pytorch3d=False\n",
7677
"try:\n",
7778
" import pytorch3d\n",
7879
"except ModuleNotFoundError:\n",
7980
" need_pytorch3d=True\n",
8081
"if need_pytorch3d:\n",
81-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
82-
" # We try to install PyTorch3D via a released wheel.\n",
83-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
84-
" version_str=\"\".join([\n",
85-
" f\"py3{sys.version_info.minor}_cu\",\n",
86-
" torch.version.cuda.replace(\".\",\"\"),\n",
87-
" f\"_pyt{pyt_version_str}\"\n",
88-
" ])\n",
89-
" !pip install fvcore iopath\n",
82+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
83+
" version_str=\"\".join([\n",
84+
" f\"py3{sys.version_info.minor}_cu\",\n",
85+
" torch.version.cuda.replace(\".\",\"\"),\n",
86+
" f\"_pyt{pyt_version_str}\"\n",
87+
" ])\n",
88+
" !pip install fvcore iopath\n",
89+
" if sys.platform.startswith(\"linux\"):\n",
90+
" print(\"Trying to install wheel for PyTorch3D\")\n",
9091
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
91-
" else:\n",
92-
" # We try to install PyTorch3D from source.\n",
93-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
92+
" pip_list = !pip freeze\n",
93+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
94+
" if need_pytorch3d:\n",
95+
" print(f\"failed to find/install wheel for {version_str}\")\n",
96+
"if need_pytorch3d:\n",
97+
" print(\"Installing PyTorch3D from source\")\n",
98+
" !pip install ninja\n",
99+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
94100
]
95101
},
96102
{

‎docs/tutorials/implicitron_volumes.ipynb‎

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -66,25 +66,31 @@
6666
"import os\n",
6767
"import sys\n",
6868
"import torch\n",
69+
"import subprocess\n",
6970
"need_pytorch3d=False\n",
7071
"try:\n",
7172
" import pytorch3d\n",
7273
"except ModuleNotFoundError:\n",
7374
" need_pytorch3d=True\n",
7475
"if need_pytorch3d:\n",
75-
" if torch.__version__.startswith(\"2.2.\") and sys.platform.startswith(\"linux\"):\n",
76-
" # We try to install PyTorch3D via a released wheel.\n",
77-
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
78-
" version_str=\"\".join([\n",
79-
" f\"py3{sys.version_info.minor}_cu\",\n",
80-
" torch.version.cuda.replace(\".\",\"\"),\n",
81-
" f\"_pyt{pyt_version_str}\"\n",
82-
" ])\n",
83-
" !pip install fvcore iopath\n",
76+
" pyt_version_str=torch.__version__.split(\"+\")[0].replace(\".\", \"\")\n",
77+
" version_str=\"\".join([\n",
78+
" f\"py3{sys.version_info.minor}_cu\",\n",
79+
" torch.version.cuda.replace(\".\",\"\"),\n",
80+
" f\"_pyt{pyt_version_str}\"\n",
81+
" ])\n",
82+
" !pip install fvcore iopath\n",
83+
" if sys.platform.startswith(\"linux\"):\n",
84+
" print(\"Trying to install wheel for PyTorch3D\")\n",
8485
" !pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/download.html\n",
85-
" else:\n",
86-
" # We try to install PyTorch3D from source.\n",
87-
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
86+
" pip_list = !pip freeze\n",
87+
" need_pytorch3d = not any(i.startswith(\"pytorch3d==\") for i in pip_list)\n",
88+
" if need_pytorch3d:\n",
89+
" print(f\"failed to find/install wheel for {version_str}\")\n",
90+
"if need_pytorch3d:\n",
91+
" print(\"Installing PyTorch3D from source\")\n",
92+
" !pip install ninja\n",
93+
" !pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'"
8894
]
8995
},
9096
{

0 commit comments

Comments
 (0)