如果报错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();
2077

被折叠的 条评论
为什么被折叠?



