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多倍。



