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 都试了一遍, 都有不同的编译报错。

以下是一些安装报错:
pytorch3d安装记录_第1张图片
pytorch3d安装记录_第2张图片

后来想到可能是cuda版本与torch不匹配, 我环境中的torch是1.13.1+cu117版本, 用的cuda版本是11.5, 改成11.7之后就能成功编译安装吗, 并且使用没有问题了。

成功安装截图:
pytorch3d安装记录_第3张图片

环境信息:

Ubuntu 22.04
gcc 11.4
python 3.8.20
CUDA 11.7
torch 1.13.1+cu117

你可能感兴趣的:(3D目标检测,Python,pytorch,人工智能,pytorch3d,3D视觉,CV)