j0seo
commented
1 year ago Hi, @scatterp2 Can you try the following setup for Colab? I'm still preparing the Colab demo, but I confirmed this env. setup code works well.
import sys
import torch
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
!git clone https://github.com/j0seo/3DFuse_Colab
!mv 3DFuse_Colab/* ./
!pip install -r requirements.txt
!mkdir weights
!wget -P ./weights https://huggingface.co/jyseo/3DFuse_weights/resolve/main/models/3DFuse_sparse_depth_injector.ckpt
!pip install pytorch-lightning==1.8.3.post0
!pip install 'prettytable<1'
!pip install 'setuptools==62.3.2'
!pip install 'jedi>=0.16'
scatterp2
I started work on a google colab anyone got any idea how to finalize this ? https://colab.research.google.com/drive/1Sv5Z1uVM5ZCGJdJSc67qGzEnyA3mlpeF?authuser=3#scrollTo=lzVAqpGLoESp