news 2026/8/16 13:35:20

【真实踩坑记录】Windows + PyTorch Geometric 复现 PointNet++ 的完整配环境过程

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
【真实踩坑记录】Windows + PyTorch Geometric 复现 PointNet++ 的完整配环境过程

本文不讲 PointNet++ 原理,也不讲网络结构
只记录一次真实的:在 Windows 上,从零开始配环境并成功跑通 PointNet++ 官方示例的全过程

如果你也在Windows + Conda + PyTorch Geometric下复现点云网络,这篇文章能帮你少走很多弯路。


一、复现目标说明

我的目标非常明确:

  • 操作系统:Windows 11

  • 使用 PyTorch Geometric(PyG)

  • 不自己写 PointNet++

  • 直接复现官方能跑的 PointNet++ 示例

  • 能正常训练、测试,不追求速度和精度


二、最开始的错误方向:尝试老项目

一开始我尝试的是:

  • 老版 PointNet++

  • torch-points3d 等 2020 年左右项目

遇到的问题

  • requirements.txt中大量包pip 已无法安装

  • 常见报错包括:

    • antlr4-python3-runtime==4.8找不到

    • gql==0.2.0不存在

    • omegaconf==2.0.x与新 pip 冲突

    • numpy / distutils / MSVC 编译错误

结论

老点云项目在 2025 年 Windows + 新 PyTorch 环境下,几乎不可直接复现

继续死磕只会无穷无尽修依赖。


三、关键转折:确认官方推荐路径

后来确认了一点非常重要的事实:

PointNet++ 已经被 PyTorch Geometric 官方实现并维护

不需要:

  • 自己写 CUDA

  • 自己写训练流程

  • 自己配一堆老依赖


四、创建 Conda 环境 & 基础依赖

conda create -n pyg-pointnetpp python=3.10 conda activate pyg-pointnetpp

安装 PyTorch(CUDA 版本,按自己显卡选):

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

安装 PyG(官方方式):

pip install torch-geometric pip install torch-scatter torch-sparse torch-cluster torch-spline-conv

验证 PyG 是否可用:

python -c "import torch_geometric; print(torch_geometric.__version__)"

能正常输出版本号即说明环境基础正确


五、克隆 PyTorch Geometric 官方仓库(重要)

git clone https://github.com/pyg-team/pytorch_geometric.git

实际遇到的问题

  • 多次出现:

    • RPC failed

    • Recv failure: Connection was reset

  • 浅克隆也失败

最终原因

网络 / GitHub 访问不稳定

在网络正常后重新克隆,成功。


六、确认 PointNet++ 示例位置

成功克隆后,关键目录是:

pytorch_geometric/examples/ ├── pointnet2_classification.py └── pointnet2_segmentation.py

⚠️ 重要认知:

PointNet++ 不在 torch_geometric.nn 里

它是一个example 级别模型

所以不要尝试:

from torch_geometric.nn import PointNet2Classification # ❌


七、第一次运行:数据集直接报错

运行:

python examples/pointnet2_classification.py

出现错误:

zipfile.BadZipFile: File is not a zip file

原因

  • ModelNet10 自动下载过程中断

  • zip 文件损坏

解决方式

删除损坏文件:

del data/ModelNet10/ModelNet10.zip

重新运行脚本,让 PyG 重新下载并解压。

成功标志:

data/ModelNet10/raw/ModelNet10/ ├── chair ├── table ├── sofa ...

八、最终状态确认

到这一步为止:

  • Conda 环境正确

  • PyTorch / PyG 正常

  • PointNet++ 示例能跑

  • 数据集正确

  • 训练 & 测试流程完整

👉复现目标达成


版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/15 3:48:38

Seal视频下载器如何通过Material You动态色彩实现个性化视觉革命

Seal视频下载器如何通过Material You动态色彩实现个性化视觉革命 【免费下载链接】Seal 🦭 Video/Audio Downloader for Android, based on yt-dlp, designed with Material You 项目地址: https://gitcode.com/gh_mirrors/se/Seal 在现代Android生态系统中&…

作者头像 李华
网站建设 2026/8/15 21:59:21

OBS多平台直播实战手册:精通高效推流配置

OBS多平台直播实战手册:精通高效推流配置 【免费下载链接】obs-multi-rtmp OBS複数サイト同時配信プラグイン 项目地址: https://gitcode.com/gh_mirrors/ob/obs-multi-rtmp obs-multi-rtmp作为OBS Studio的专业级多平台推流解决方案,彻底改变了创…

作者头像 李华
网站建设 2026/8/16 2:02:10

JUCE音频开发终极指南:从零到专业级插件工程师

JUCE音频开发终极指南:从零到专业级插件工程师 【免费下载链接】JUCE 项目地址: https://gitcode.com/gh_mirrors/juc/JUCE 想要快速掌握音频插件开发技能?JUCE框架正是你实现这一目标的完美工具。作为业界领先的跨平台音频应用开发框架&#xf…

作者头像 李华
网站建设 2026/8/16 12:22:20

Agent Zero多语言界面终极配置指南:零代码解锁全球团队协作

Agent Zero多语言界面终极配置指南:零代码解锁全球团队协作 【免费下载链接】agent-zero Agent Zero AI framework 项目地址: https://gitcode.com/GitHub_Trending/ag/agent-zero Agent Zero AI框架提供全面的多语言界面配置功能,让用户无需编写…

作者头像 李华