A40服务器splatam环境搭建

0 准备

diff-gaussian-rasterization-w-depth.zip, CUDA Toolkit 12.1， 百度网盘
谷歌扩展：标签修改精灵 - 修改标签名称和图标

1 安装CUDA

1.1 下载CUDA

根据图片下载cuda_12.1.0_530.30.02_linux.run文件。

1.2 安装CUDA

使用以下命令按照CUDA：

sudoshcuda_12.1.1_530.30.02_linux.run

注意：

• CUDA Toolkit [x] (这是必须的，保持选中)
• NVIDIA Driver [ ] (取消选中这个！ 因为你已经有驱动了，重新安装可能会导致问题)
• CUDA Samples [ ] (可选，可以不装)
• CUDA Demo Suite [ ] (可选，可以不装)

1.3 设置环境变量

echo'export CUDA_HOME=/usr/local/cuda'>>~/.bashrcecho'export PATH=$CUDA_HOME/bin:$PATH'>>~/.bashrcecho'export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH'>>~/.bashrc

source~/.bashrc

关闭终端

1.4 验证CUDA

nvcc--version

2 安装diff-gaussian-rasterization-w-depth

pipinstalldiff-gaussian-rasterization-w-depth-cb65e4b86bc3bd8ed42174b72a62e8d3a3a71110.zip --no-build-isolation

3 安装其他库

sudodnfinstallnano

sudonano~/.bashrc

# 安装natsortpipinstallnatsort# 安装korniapipinstallkornia# 安装pytorch_msssimpipinstallpytorch_msssim pipinstallwandb

4 整理splatam工程

4.1 数据准备

新建TUM_RGBD文件，解压rgbd_dataset_freiburg3_walking_static.tgz文件

• data
  - TUM_RGBD
  - rgbd_dataset_freiburg3_walking_static

4.2 修改文件

路径：configs/tum/splatam.py
添加：‘freiburg3_walking_static’

scenes=["freiburg3_walking_static","freiburg1_desk2","freiburg1_room","freiburg2_xyz","freiburg3_long_office_household"]

添加：freiburg3_walking_static.yaml
路径：configs/data/TUM/freiburg3_walking_static.yaml

dataset_name:'tum'camera_params: image_height:480image_width:640fx:535.4fy:539.2cx:320.1cy:247.6crop_edge:8png_depth_scale:5000.0

修改datasets/init.py文件，将_init_.py改成__init__.py
修改configs/tum/splatam.py中的use_wandb，改为False