关于pytorch3d的安装

诚威_lol_中大努力中2024-12-27 15:18

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 .的

4、参考文章3：Prebuilt wheels provided via 3rd party repository · facebookresearch/pytorch3d · Discussion #1752 (github.com)

这个是文章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/

下载到本地去安装即可。