type
status
date
slug
summary
tags
category
icon
password
Last edited time
Jan 15, 2025 12:47 PM
安装pointnet2_ops_lib需要调整CUDA，Pytorch，GCC版本时踩的坑
📝 主旨内容
环境准备Setup
We implement this repo with the following environment:
- Ubuntu 18.04
- Python 3.8
- Pytorch 1.9.0
- CUDA 11.3
Install the other package via:
其实不一定要11.3，11.3~11.8都是可以的，下面以11.3为例(可能需要调整CUDA和GCC版本)，然后requirements.txt缺少一些依赖，安装pointnet2_ops_lib会超时
CUDA11.3安装
CUDA版本管理(切换到11.3)
Pytorch安装(需要对应cuda版本)
安装pointnet2_ops_lib(离线安装)
GCC版本管理(可选，上面报错了换就行)
导入checkpoints
导入数据集
预处理数据集
训练Train
训练并测试双库版本并保存该功能以进行 UFF 训练：
训练 UFF：
使用以下命令训练和测试完整设置：
注意：如果设置了
--method_name DINO 或 --method_name Point_MAE ，请同时设置 --memory_bank single 。ModuleNotFoundError: No module named 'torch._six’
Ninja is required to load C++ extensions
🤗 总结归纳
📎 参考文章
- 作者:ziuch
- 链接:https://ziuch.com/article/M3DM-pointnet2-ops-install-guide
- 声明:本文采用 CC BY-NC-SA 4.0 许可协议，转载请注明出处。
相关文章
