Anaconda下PyTorch3D模块安装
Cleo Lv2
 2021-12-10 18:17:00    

PyTorch3D是PyTorch对几何方向处理的封装库。笔者曾在安装时出错，这里记录一个比较容易成功的安装方法。

从Gitee上克隆仓库并安装:

1
2
$ git clone https://gitee.com/starstar365/pytorch3d.git
$ cd pytorch3d && pip install -e .

这个版本并非最新的版本，经过实验，在1080Ti、Titan XP、3090和A4000上都能成功安装，不乏cuda版本为10.1和11.1。