pytorch3d installation on colab taking too long
Closed this issue · 5 comments
I am running this on colab:
import os
import sys
import torch
need_pytorch3d=False
try:
import pytorch3d
except ModuleNotFoundError:
need_pytorch3d=True
if need_pytorch3d:
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'
This results in a correct installation but it takes every time around 25 minutes to complete, the torch version is 2.4.0+cu121, is there any way to speed up the process? Maybe upgrade to an earlier version? I've tried a few, even changing torch versions, but they don't seem to work.
Thanks, I used this code taken from the updated tutorials:
import os
import sys
import torch
import subprocess
need_pytorch3d=False
try:
import pytorch3d
except ModuleNotFoundError:
need_pytorch3d=True
if need_pytorch3d:
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 iopath
if sys.platform.startswith("linux"):
print("Trying to install wheel for PyTorch3D")
!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:
print(f"failed to find/install wheel for {version_str}")
if need_pytorch3d:
print("Installing PyTorch3D from source")
!pip install ninja
!pip install 'git+https://github.com/facebookresearch/pytorch3d.git@stable'
This gave an error, fixed by adding !pip install fvcore after the line !pip install iopath. Now it works and is fast.
Thanks for that. I know why it's happening. The wheel is version 0.7.7 but the code is corresponding to version 0.7.8 in which we newly do not depend on fvcore. We can fix by updating the wheel.
Should now be done
And it's now pyt2.4.1; also done.