1、参考文章1:windows 安装 PyTorch3D 详细指南 - 哔哩哔哩 (bilibili.com)
这篇文章巨好
2、参考文章2:pytorch3d/INSTALL.md at main · facebookresearch/pytorch3d (github.com)
这个是官方的install教程
3、我是直接从github上面下载pytorch3d.zip,然后解压后,在里面pip install -e .的
这个是文章1中推荐的一个讨论,非常方便,里面包含了所有需要的.whl轮子文件,直接从轮子安装!
提前查询好Pytorch和CUDA版本
python -c "import torch; print(torch.__version__)"
用这个命令可以查出,比如Pytorch2.0.1CU117(就是CUDA11.7)
同时python比如是3.9的
linux是x86_64的
从这里找.whl文件:miropsota.github.io/torch_packages_builder/pytorch3d/
下载到本地去安装即可。