【linux】安装pytorch3d

写在开头

不要看网上的博客安装，直接参考官方安装文档去安装。

坑1：安装pytorch3d后torch.cuda用不了

使用下面的命令安装后

conda install pytorch3d

会提示安装下面的包，注意pytorch和torchvision都是cpu版本的，这样就会覆盖我们原本cuda版本的pytorch和torchvision,导致之后发现torch.cuda.is_available()返回False

解决办法1：所有都用conda安装

开始安装torch时就用conda安装，不要用pip安装
之后安装pytorch3d也用conda安装

解决办法2：从源码安装pytorch3d

看似最复杂，其实最有效
第一步：git clone
第二步：编译安装

git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d && pip install -e .