如果报错no module named pointops_cuda
说明pointops的包没有安装,
切换到该setup.py所在的文件夹下,即 libs/pointops
输入:python3 setup.py install

若一切顺利,则能成功完成安装,

若有报错,目前遇到两个
1:

AT_CHECK,找到pointops/src中,点开每个.cpp文件将其中的

#define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")
#define CHECK_CONTIGUOUS(x) AT_CHECK(x.is_contiguous(), #x, " must be contiguous ")
将AT换为TORCH,即TORCH_CHECK
2:

THC/THC.h相关的,找到pointops/src中,点开每个.cpp文件将其中的
//     cudaStream_t stream = THCState_getCurrentStream(state);
替换为:cudaStream_t stream = at::cuda::getCurrentCUDAStream().stream();

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐