• Roeia Kishk's avatar
    Changed tutorials' pip searching · 31e3488a
    Roeia Kishk authored
    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
    31e3488a
fit_textured_volume.ipynb 19.9 KB