type
status
date
slug
summary
tags
category
icon
password
Last edited time
Aug 12, 2024 04:55 PM
😀
安装pointnet2_ops_lib需要调整CUDA，Pytorch，GCC版本时踩的坑
notion image

📝 主旨内容

环境准备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安装

NVIDIA CUDA Toolkit 12.1 Downloads
Get the latest feature updates to NVIDIA's compute stack, including compatibility support for NVIDIA Open GPU Kernel Modules and lazy loading support.
NVIDIA CUDA Toolkit 12.1 Downloads
https://developer.nvidia.com/cuda-downloads
NVIDIA CUDA Toolkit 12.1 Downloads

CUDA版本管理(切换到11.3)

Pytorch安装(需要对应cuda版本)

安装pointnet2_ops_lib(离线安装)

GCC版本管理(可选，上面报错了换就行)

导入checkpoints

导入数据集

预处理数据集

训练Train

训练并测试双库版本并保存该功能以进行 UFF 训练：
训练 UFF：
使用以下命令训练和测试完整设置：
注意：如果设置了 --method_name DINO 或 --method_name Point_MAE ，请同时设置 --memory_bank single 。

🤗 总结归纳

 

📎 参考文章

 
相关文章
ChatPaper-Click——让GPT4帮你解读论文
Xboard机场面板及Xrayr后端搭建
自建图床及随机图片API
ChatpaperV2项目重启(装死中)
Github个人主页/论文项目教程模板
白嫖GPT3.5API
菜鸟春招实习笔试题解(0530)工业异常检测——重构(Reconstruction)
Loading...