ubuntu20.04安装OpenPcdet,CUDA版本11.8,显卡4090

本文参考这2篇文章的内容:https://blog.csdn.net/jin15203846657/article/details/122735375#comments_25352667
https://zhuanlan.zhihu.com/p/642158810

记录了自己安装OpenPcdet的过程。

OpenPcdet的安装需要cuda和pytorch版本严格关联。本例的CUDA版本:11.8,pytorch版本:1.13.0,torchaudio版本:0.13.0,torchvision版本:0.14.0。

创建并激活虚拟环境

bash 复制代码
conda create -n pcdet python=3.9 -y
conda activate pcdet

拉取pcdet源码

bash 复制代码
git clone https://github.com/open-mmlab/OpenPCDet.git

安装spconv,具体安装命令参考https://github.com/traveller59/spconv

本文安装命令如下

bash 复制代码
pip install spconv-cu118 

安装torch相关,需要指定版本,可以参考https://pytorch.org/get-started/previous-versions/

bash 复制代码
pip install torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

安装完指定版本的torch相关包之后,cd到pcdet根目录下,安装requirements.txt

bash 复制代码
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple/

运行命令

bash 复制代码
python setup.py develop
pip install open3d

测试OpenPcdet是否安装成功,先下载测试的bin文件和模型文件,运行下面的测试命令

bash 复制代码
# 注意自己的OpenPCDet源码路径哈
cd /home/user/OpenPCDet/tools
python demo.py --cfg_file /home/user/OpenPCDet/tools/cfgs/kitti_models/pointpillar.yaml --ckpt /home/user/OpenPCDet/tools/pointpillar_7728.pth --data_path /home/user/OpenPCDet/tools/000001.bin

成功运行后会显示如下

可能出现的报错,参考这篇文章https://blog.csdn.net/weixin_44577224/article/details/132123873?spm=1001.2014.3001.8078#comments_35431638

1、ModuleNotFoundError: No module named 'av2'

解决办法:pip install av2

2、ModuleNotFoundError: No module named 'kornia'

解决办法:pip install kornia==0.6.5

这里如果不指定版本的话,后续运行也会报错

相关推荐
百锦再19 分钟前
[特殊字符] Python在CentOS系统执行深度指南
开发语言·python·plotly·django·centos·virtualenv·pygame
张子夜 iiii28 分钟前
4步OpenCV-----扫秒身份证号
人工智能·python·opencv·计算机视觉
潮汐退涨月冷风霜1 小时前
数字图像处理(1)OpenCV C++ & Opencv Python显示图像和视频
c++·python·opencv
九章云极AladdinEdu6 小时前
超参数自动化调优指南:Optuna vs. Ray Tune 对比评测
运维·人工智能·深度学习·ai·自动化·gpu算力
酷飞飞8 小时前
Python网络与多任务编程:TCP/UDP实战指南
网络·python·tcp/ip
研梦非凡9 小时前
ICCV 2025|从粗到细:用于高效3D高斯溅射的可学习离散小波变换
人工智能·深度学习·学习·3d
数字化顾问9 小时前
Python:OpenCV 教程——从传统视觉到深度学习:YOLOv8 与 OpenCV DNN 模块协同实现工业缺陷检测
python
学生信的大叔10 小时前
【Python自动化】Ubuntu24.04配置Selenium并测试
python·selenium·自动化
诗句藏于尽头11 小时前
Django模型与数据库表映射的两种方式
数据库·python·django
通街市密人有11 小时前
IDF: Iterative Dynamic Filtering Networks for Generalizable Image Denoising
人工智能·深度学习·计算机视觉