3D Gaussian Splatting环境配置和运行

本文详细指导如何为中国学生在中国环境下配置3DGaussianSplatting项目,包括下载库、创建虚拟环境、安装特定版本PyTorch和子模块,以及运行训练和渲染脚本。
摘要由CSDN通过智能技术生成

1. 下载3dgs到本地

下载3dgs:

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

进入3dgs的目录里:

cd gaussian-splatting

2. 创建对应的虚拟环境

(1)修改environment.yml文件

将environment.yml文件最后三行注释掉:

#  - pip:
#    - submodules/diff-gaussian-rasterization
#    - submodules/simple-knn

原因是,在第(2)步使用conda命令根据environment.yml文件创建虚拟环境时,安装的pytorch是使用conda install 命令来安装的,实际上并不能成功安装带有gpu的pytorch,而这两个模块的安装依赖于带gpu的pytorch。
如果不注释掉这三行,直接安装会报错!

(2)创建虚拟环境

创建虚拟环境:

conda env create --file environment.yml

激活虚拟环境:

conda activate gaussian_splatting

(3)单独安装pytorch

根据yml文件,可以知道需要安装pytorch=1.12.1 cudatoolkil=11.6的pytorch,使用pip安装,才能成功装上:

pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116

(4)安装两个submodule

现在就差两个submodule没安装了

pip install submodules/diff-gaussian-rasterization
pip install submodules/simple-knn

到此环境配置就结束啦!

3. 运行

这部分没什么难度啦,就参照README.md就行

python train.py -s <path to COLMAP or NeRF Synthetic dataset> --eval # Train with train/test split
python render.py -m <path to trained model> # Generate renderings
python metrics.py -m <path to trained model> # Compute error metrics on renderings
  • 5
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值