当前位置: 首页 > news >正文

3DGS训练速度实测:用Ubuntu 20.04 + RTX 4090跑官方数据集,看看30K轮要多久?

3DGS训练速度实测:RTX 4090在Ubuntu 20.04上的性能表现

当3D Gaussian Splatting(3DGS)技术逐渐成为实时辐射场渲染的新标杆,许多开发者和研究者最关心的问题之一就是:在自己的硬件配置上,训练一个完整模型究竟需要多长时间?本文将基于Ubuntu 20.04系统和RTX 4090显卡,对官方Truck数据集进行从环境搭建到完整训练的全程实测,记录关键性能指标,为考虑采用3DGS技术的团队提供真实的硬件参考数据。

1. 测试环境配置与准备工作

在开始性能测试前,一套稳定且符合要求的软件环境是基础。我们的测试平台采用以下配置:

  • 硬件配置

    • GPU:NVIDIA RTX 4090 (24GB GDDR6X)
    • CPU:AMD Ryzen 9 7950X
    • 内存:64GB DDR5
    • 存储:2TB NVMe SSD
  • 软件环境

    • 操作系统:Ubuntu 20.04.6 LTS
    • CUDA版本:11.8
    • 驱动版本:520.61.05
    • Python环境:Anaconda with Python 3.9

关键组件安装顺序与注意事项

  1. CUDA Toolkit安装

    wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run sudo sh cuda_11.8.0_520.61.05_linux.run

    提示:安装CUDA时建议选择不安装驱动,如果系统已有较新驱动,避免版本冲突。

  2. 系统环境变量配置

    echo 'export PATH=/usr/local/cuda-11.8/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc
  3. 依赖库版本对照表

    组件名称推荐版本备注
    gcc/g++9.x过高版本可能导致兼容性问题
    CMake≥3.24需支持C++17特性
    Eigen3.4.0部分功能需要此版本
    OpenCV4.9.0需包含contrib模块

2. 3DGS环境搭建实战

2.1 源码获取与虚拟环境创建

官方仓库的克隆需要特别注意--recursive参数,确保所有子模块正确下载:

git clone https://github.com/graphdeco-inria/gaussian-splatting --recursive cd gaussian-splatting

创建专用虚拟环境可避免系统Python环境被污染:

conda env create -f environment.yml conda activate gaussian_splatting

2.2 编译优化与疑难解决

在实际搭建过程中,可能会遇到几个典型问题:

  • gcc版本冲突

    sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 90 sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-9 90
  • 可视化组件编译失败: 进入SIBR_viewers目录后,针对Ubuntu 20.04需要额外执行:

    git checkout fossa_compatibility cmake -Bbuild . -DCMAKE_BUILD_TYPE=Release cmake --build build -j24 --target install

注意:编译过程会占用大量内存,建议关闭其他内存密集型应用。遇到错误时,通常与依赖版本有关,需根据具体错误信息调整。

3. 训练过程与性能指标分析

3.1 数据集准备与训练启动

使用官方提供的Truck数据集可以排除数据质量问题对性能测试的影响:

python train.py -s data/truck/ -m data/truck/output

关键参数解析

  • -s:指定源数据路径
  • -m:模型输出路径
  • -i:可指定迭代次数(默认为30,000)

3.2 实时性能监控数据

在RTX 4090上的训练过程中,我们记录了以下关键指标:

迭代区间耗时(秒)GPU显存占用(GB)GPU利用率(%)
0-1K4218.398
1K-5K19519.199
5K-10K38019.798
10K-20K75220.297
20K-30K110820.596

性能观察结论

  1. 初期迭代(0-1K)速度最快,随着高斯点数量增加,每轮迭代耗时逐渐增加
  2. GPU显存占用稳定在20GB左右,RTX 4090的24GB显存完全满足需求
  3. GPU利用率始终保持在95%以上,说明计算资源得到充分利用

3.3 质量评估指标变化

训练过程中记录的渲染质量指标变化趋势:

迭代次数PSNR(dB)SSIM点云数量(百万)
1,00021.340.7810.32
5,00024.560.8520.87
10,00026.780.8911.23
20,00028.120.9131.45
30,00028.930.9261.52

从数据可以看出,前10,000轮迭代质量提升最为显著,后期趋于平缓。实际应用中可根据需求适当调整总迭代次数。

4. 可视化效果与实用建议

4.1 实时查看器使用技巧

编译安装的可视化工具位于SIBR_viewers/install/bin目录,启动命令为:

./SIBR_gaussianViewer_app -m /path/to/output

查看器功能快捷键

  • F1:显示/隐藏UI面板
  • WASD:场景漫游
  • 鼠标拖动:视角旋转
  • 空格:暂停/继续动画

4.2 针对不同硬件的调优建议

根据本次测试经验,针对不同配置的用户给出以下建议:

低显存显卡(<12GB)

  • 降低--resolution参数值
  • 使用--densify_until_iter提前停止点云增密
  • 考虑使用--reduce_after_iter主动减少点云数量

多GPU系统

  • 目前官方代码未原生支持多GPU,可尝试手动拆分数据集
  • 使用NCCL库实现分布式训练需要修改底层CUDA代码

训练中断恢复

python train.py -s data/truck/ -m data/truck/output --resume

重要提示:定期备份point_cloud目录下的.ply文件,这些是训练过程中的检查点。

在实际项目中,我们发现3DGS对输入数据质量非常敏感。使用自采集数据时,建议先用Colmap进行高质量稀疏重建,确保相机参数估计准确。训练过程中如果发现PSNR提升缓慢,很可能是初始点云质量不佳,需要检查前期处理流程。

http://www.jsqmd.com/news/753657/

相关文章:

  • 5分钟掌握Xiaomusic:让小爱音箱变身智能音乐播放器的终极指南
  • 从一道CTF题学PHP安全:如何用php://filter绕过文件包含过滤(以BUUOJ Secret File为例)
  • Prompt Engineering与LangChain实战:从零构建AI应用开发指南
  • Python 统计学基础与高级应用
  • 计算机网络复习(第五章):传输层
  • 大语言模型文本中的文化特征与作者风格识别技术
  • 智能体框架如何让大语言模型实现系统性思考与复杂任务处理
  • 2026上海卫生间免砸砖防水、外墙、地下室、楼顶渗漏+彩钢瓦、阳光房隔热 本地专业防水公司TOP5权威推荐(2026年5月本地最新深度调研) - 企业资讯
  • Linux 的 tee 命令
  • 量子最优控制的鲁棒性优化与离散化误差修正
  • 智能搜索DeepAgent笔记
  • 别再只用D435i测深度了!解锁它在ROS Melodic/Noetic下的三种高阶玩法
  • 如何随时随地访问你的“进程”?
  • 前后端分离架构中怎么防止 JWT 被 XSS 攻击窃取?
  • WorkshopDL:3分钟学会免费下载Steam创意工坊模组的终极方案
  • d2s-editor:暗黑破坏神2存档编辑难题的终极解决方案
  • XML数据解析与节点配置实操案例教程
  • KingFusion|最近开发调试中遇到的几个问题及解决办法(2)
  • Legacy-iOS-Kit:如何用开源技术让经典iOS设备重获新生?
  • 2026包头卫生间免砸砖防水、外墙、地下室、楼顶渗漏+彩钢瓦、阳光房隔热 本地专业防水公司TOP5权威推荐(2026年5月本地最新深度调研) - 企业资讯
  • 别再只会用!pwd了:深度解析Kaggle Notebook的目录结构与高效工作流设计
  • 2026厦门卫生间免砸砖防水、外墙、地下室、楼顶渗漏+彩钢瓦、阳光房隔热 本地专业防水公司TOP5权威推荐(2026年5月本地最新深度调研) - 企业资讯
  • 2026青岛卫生间免砸砖防水、外墙、地下室、楼顶渗漏+彩钢瓦、阳光房隔热 本地专业防水公司TOP5权威推荐(2026年5月本地最新深度调研) - 企业资讯
  • 2026东营卫生间免砸砖防水、外墙、地下室、楼顶渗漏+彩钢瓦、阳光房隔热 本地专业防水公司TOP5权威推荐(2026年5月本地最新深度调研) - 企业资讯
  • 解锁新体验:如何用Video Speed Controller重塑你的视频观看习惯
  • 基于STM32H743与LoRa的诺基亚E63独立通信改造方案
  • 均线交易执行清单
  • 对求导法则和积分法则的一些思考
  • 26山大软院创新实训--MarketClaw(三)
  • [理论篇-9]Skill系统与能力封装