pytorch3d安装记录
官方安装教程: https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md
通过pip 或conda 可以很容易安装上预编译好的包, 安装过程不会报错, 但是使用的时候就会报各种错误 ,原因是预编译好的包跟自己的环境不一定匹配, 比如CUDA版本, torch版本等。
推荐从源码安装。
我是直接下载的源码安装, 但是执行python setup.py install 时有各种编译报错, 其实怀疑是gcc版本问题, gcc 版本 从gcc-7, gcc-8, gcc-9, gcc-10, gcc-11 都试了一遍, 都有不同的编译报错。
以下是一些安装报错:
后来想到可能是cuda版本与torch不匹配, 我环境中的torch是1.13.1+cu117
版本, 用的cuda版本是11.5, 改成11.7之后就能成功编译安装吗, 并且使用没有问题了。
成功安装截图:
环境信息:
Ubuntu 22.04
gcc 11.4
python 3.8.20
CUDA 11.7
torch 1.13.1+cu117