栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Torch3d windows10安装踩坑

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Torch3d windows10安装踩坑

https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md
之前需要配好gpu版的torch和torchvision
环境:pytorch 1.9 cuda 11.1 cub1.9.10-1 MSVC14.27.29110 torch3d 0.6.0
Linux下可以直接pip,但现在这个包对windows不是很友好,需要下源码自己编译。
前面的运行时配置按文档来没问题,到setup install的时候,默认是ninja这个python包编译,我用这个包会有各种问题,然后按照issue的答案可以通过set PYTORCH3D_NO_NINJA=1把它禁掉,然后会选择MSVC作为编译器。
给出的说明是基本不支持vs2019之前的编译器了,但是最新的2019MSVC版本在14.29往上,这会导致nvcc编译cu文件的时候出错,具体来说编译到sigmoid_alpha_blend.cu或者ball_query.cu的时候就会弹一大堆语法error。需要把MSVC降到14.27,最好的方法是在vs installer里面卸了重新装一次,装的时候只勾选win10 kits和msvc14.27。然后需要在环境变量里加vc,win10 kits的include和lib,这部分网上有参考,install的时候缺什么搜索引擎拷贝下error信息就有答案。
中间编译的时候有个iou_box3d.cuh的文件,会无法识别not。定位到对应行改成 ! 就行。
测试下tutorial1,cpu版每秒大概迭代3次,gpu版快了20多倍。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/313545.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号