本文介绍: 编译Pointnet2.PyTorch仓库中的setup.py时报THCTHC.h: No such file or directory。原因是THC.h在pytorch 1.11版本中被移除了，需要修改代码。部分仓库中setup.py文件在子文件夹中，需要使用pip install 从子文件夹中直接安装某个分支。比如上面这个PointNet2.Pytorch仓库的setup.py在子文件夹。
问题1描述
编译Pointnet2.PyTorch仓库中的setup.py时报THCTHC.h: No such file or directory。
问题解决
原因是THC.h在pytorch 1.11版本中被移除了，需要修改代码。仓库issues给除了解决方法：
//Comment Out /src/ball_query.cpp group_points.cpp interpolate.cpp pointnet2_api.cpp sampling.cpp
把上面列出的文件的以下行注释掉
//#include <THE/THC.h>
//extern THCState *state;
//cudaStream_t stream = THCState_getCurrentStream(state);
改为
#include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAEvent.h>
cudaStream_t stream = at::cuda::getCurrentCUDAStream();
然后
replace the AT_CHECK with TORCH_CHECK in the ball_query.cpp
问题2描述
部分仓库中setup.py文件在子文件夹中，需要使用pip install 从子文件夹中直接安装某个分支
解决方案
比如上面这个PointNet2.Pytorch仓库的setup.py在子文件夹pointnet2中，可以这样安装branch分支：
pip install git+https://github.com/sshaoshuai/Pointnet2.PyTorch.git@master#subdirectory=pointnet2
原文地址:https://blog.csdn.net/qq_41035283/article/details/128615334
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如若转载，请注明出处：http://www.7code.cn/show_45892.html
如若内容造成侵权/违法违规/事实不符，请联系代码007邮箱：suwngjj01@126.com进行投诉反馈，一经查实，立即删除！
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
