pytorch3d安装记录

打印 上一主题 下一主题

主题 842|帖子 842|积分 2526

官方安装教程: 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之后就能乐成编译安装吗, 并且使用没有题目了。
乐成安装截图:

环境信息:
  1. Ubuntu 22.04
  2. gcc 11.4
  3. python 3.8.20
  4. CUDA 11.7
  5. torch 1.13.1+cu117
复制代码
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

x
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

悠扬随风

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表