Pytorch3d installation tip

[willxxy]

Hello again,

I had some trouble installing pytorch3d for cuda 11.6 from the instructions so I found a very useful repo that shows how to modify the pytorch3d version with a custom pytorch and cuda version.

The TLDR of the issue says to use this command

pip install --extra-index-url https://miropsota.github.io/torch_packages_builder pytorch3d==<version>+<OPTIONAL_commit_hash>pt<PyTorch_version><compute_platform>

Where <compute_platform> is as in PyTorch, one of cpu, cu<CUDA_short_version> (e.g. cu121, cu118, cu102), or rocm<ROCM_version> (not supported right now).

For example, I used the following installation for my setup

pip install --extra-index-url https://miropsota.github.io/torch_packages_builder pytorch3d==0.7.5+pt1.13.1cu116

I hope this helps anyone who has trouble with this! And thank you MiroPsota for the original issue!

[YinpeiDai]

Thank you for the tips! I usually use conda to build pytorch3d, so I never encounter installment issue. Hope your experience on pip will help others~