Ubuntu22.04配置3D gaussian splatting

news2024/12/19 17:34:23

这篇博客提供了3D gaussian splatting在新安装Ubuntu上的配置过程。

1.拉仓库

2.安装显卡驱动和cuda版本

3.安装Pytorch

4.安装Pycharm和配置Python

5.安装附加依赖项（方法一）

6.安装Anaconda（方法二）

7.测试

1.拉仓库

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

2.安装显卡驱动和cuda版本

打开终端，输入命令：

nvidia-smi

上图的CUDA Version：12.4指的是最高支持cuda12.4的版本，并不是已经安装了cuda12.4的版本，输入以下指令查看cuda版本：

nvcc -V

上图表示cuda未安装，此时需要去Nvidia官网CUDA Toolkit 12.6 Update 3 Downloads | NVIDIA Developer安装CUDA Toolkit，注意不要超过上述规定的CUDA版本

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4

配置环境变量

为确保正确配置 CUDA 环境变量，编辑 ~/.bashrc 文件：

nano ~/.bashrc

在文件末尾添加以下内容：

export PATH=/usr/local/cuda-12.4/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64/stubs:${LD_LIBRARY_PATH}

然后，执行以下命令使配置生效：

source ~/.bashrc

可以通过以下命令验证 CUDA 是否安装成功：

nvcc --version

3.安装Pytorch

安装cuda版本的pytorch版本。本人显卡3090，cuda版本12.4

pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124

4.安装Pycharm和配置Python

Step 1: 安装pycharm

sudo snap install pycharm-professional --classic

进去选择中文版。

确保你已经安装了 Python。运行以下命令检查：

python3 --version

如果未安装 Python，使用以下命令安装：

sudo apt update
sudo apt install python3 python3-pip

Step 2: 打开 PyCharm

启动 PyCharm。
创建或打开一个项目。

Step 3: 配置 Python 解释器

点击菜单 File > Settings（或 PyCharm > Preferences）。
导航到 Project > Python Interpreter。
点击右侧的齿轮图标，选择 Add Interpreter。
选择 System Interpreter 或 Virtual Environment：
- System Interpreter:
  - 在下拉菜单中选择你的 Python 路径（通常是 /usr/bin/python3 或类似路径）。
- Virtual Environment:
  - 选择 New Virtualenv Environment。
  - 确保 Python 二进制文件指向正确的 Python 版本。
点击 OK，等待 PyCharm 配置环境。

5.安装附加依赖项（方法一）

# Dependencies
sudo apt install -y libglew-dev libassimp-dev libboost-all-dev libgtk-3-dev libopencv-dev libglfw3-dev libavdevice-dev libavcodec-dev libeigen3-dev libxxf86vm-dev libembree-dev

pip install plyfile

pip install opencv-contrib-python

pip install tqdm

# Project setup
cd SIBR_viewers
cmake -Bbuild . -DCMAKE_BUILD_TYPE=Release # add -G Ninja to build faster
cmake --build build -j24 --target install

安装子模块

先cd到~/code/gaussian-splatting，然后输入命令：