search

facebookresearch / pytorch3d

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
https://pytorch3d.org/
Other
8.7k stars 1.3k forks source link

Where to find pre-build wheels? #1780

Closed jasonyzhang closed 5 months ago

jasonyzhang commented 5 months ago

I have found that the easiest way to install Pytorch3D is using the pre-built wheel, e.g. pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu113_pyt1110/download.html

Is there any easy way to find which python/CUDA/pytorch version combinations there are pre-built wheels?

I also found that the CI config appears to build a large number of wheels for conda. However, none of the builds for pytorch 2.2.0 seem to be included.

bottler commented 5 months ago

We really encourage use of conda packages rather than wheels. The wheels really only exist for colab users. And I think there are no conda packages for PyTorch 2.2 because it is newer than the latest release of PyTorch3D release.

The available wheels are listed in https://github.com/facebookresearch/pytorch3d/issues/1401 , it should include the actual version needed for colab.

jasonyzhang commented 5 months ago

Thanks for the explanation and the pointers! That was very helpful