pytorch3d 安装

分享一下在Ubuntu 20.04系统上安装PyTorch3D的经验，尤其是当遵循官方教程却遇到安装困难时。PyTorch3D是由Facebook AI Research团队开发的，旨在支持与3D数据相关的深度学习任务，包括但不限于3D形状的建模、渲染和视觉理解。

准备工作

在安装PyTorch3D之前，首先需要确保您的系统已安装Python和pip。PyTorch3D对CUDA版本有特定要求，因此确认CUDA的安装路径和版本与PyTorch3D的要求相匹配是必要的步骤。

下载源代码

开始安装PyTorch3D之前，推荐从GitHub仓库下载源代码到本地。这可以通过克隆仓库或下载ZIP文件完成。下载完成后，解压（如果需要）并进入源代码目录。 然后使用pip install -e .命令，该命令会以编辑模式安装PyTorch3D。

遇到的问题

1. 无法安装
2. 下载了pytorch3d, 但是运行时候报错

解决方案

解决上述问题的关键步骤是在安装前正确设置CUDA路径。这可以通过导出CUDA_HOME环境变量来完成，如下所示：

export CUDA_HOME=/usr/local/cuda-<version>  # Replace <version> with your CUDA version
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

这里的/usr/local/cuda-x.x应替换为您的CUDA安装路径。设置正确后，重新执行安装命令pip install -e .，应能顺利完成安装且使用。

此外，我发现在安装前最后安装pytorch建议安装的cuda设置。

conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

Issacgym 使用它推荐的是python 3.7，但是pytroch3d官方文档现在只支持3.8, 3.9 or 3.10。不知道这个会不会影响，反正按照它推荐的cuda和torch version 肯定不会错。