Advertisement

Ubuntu22.04配置3D gaussian splatting

阅读量:

这篇博客提供了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版本

复制代码
>       1. wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin

>  
>       2. sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
>  
>       3. 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
>  
>       4. sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
>  
>       5. sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
>  
>       6. sudo apt-get update
>  
>       7. 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**

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

##### **Step 3: 配置 Python 解释器**

  1. 

点击菜单 **File > Settings**(或 **PyCharm > Preferences**)。

  2. 

导航到 **Project > Python Interpreter**。

  3. 

点击右侧的齿轮图标,选择 **Add Interpreter** 。

  4. 

选择 **System Interpreter** 或 **Virtual Environment** :

     * **System Interpreter** : 
       * 在下拉菜单中选择你的 Python 路径(通常是 `/usr/bin/python3` 或类似路径)。
     * **Virtual Environment** : 
       * 选择 **New Virtualenv Environment** 。
       * 确保 Python 二进制文件指向正确的 Python 版本。
  5. 

点击 **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,然后输入命令:

>
>
> pip install submodules/diff-gaussian-rasterization
>
>
>
> pip install submodules/simple-knn  
>  pip install submodules/fused-ssim
>
>

## 6.安装Anaconda(方法二)

##### **步骤 1:下载 Anaconda 安装脚本**

访问 Anaconda 下载页面 下载适用于 Linux 的安装脚本,或者直接使用 `wget` 命令:

>
>
> wget https://repo.anaconda.com/archive/Anaconda3-2023.03-Linux-x86_64.sh
>
>

##### **步骤 2:运行安装脚本**

运行下载的安装脚本:

>
>
> bash Anaconda3-2023.03-Linux-x86_64.sh
>
>

##### **步骤 3:安装过程中接受许可协议**

与 Miniconda 类似,Anaconda 安装过程会提示你接受许可协议。输入 `yes` 来继续。

##### **步骤 4:选择安装路径**

可以选择默认路径(通常是 `~/anaconda3`),或者指定一个其他目录。

##### **步骤 5:初始化 Conda**

选择 `yes` 来初始化 Conda。

##### **步骤 6:刷新 shell 配置**

与 Miniconda 一样,安装完成后,你需要刷新 shell 配置:

>
>
> source ~/.bashrc
>
>

##### **步骤 6:** 验证 Conda 是否成功安装:

>
>
> conda --version
>
>

##### **步骤 7:** 创建环境并安装依赖项

>
>
> conda env create --file environment.yml conda activate gaussian_splatting
>
>
>
> conda activate gaussian_splatting
>
>

## 7.测试

下载DTU数据集<https://drive.google.com/drive/folders/1SJFgt8qhQomHX55Q4xSvYE2C6-8tFll9>,并进行解压。

设置形参如下,运行即可。
![](https://ad.itadn.com/c/weblog/blog-img/images/2025-05-03/QVjfbxCsNzWJ8p6oYZ5AdhOMaBHw.png)

运行结果的终端显示:
![](https://ad.itadn.com/c/weblog/blog-img/images/2025-05-03/Elr5z2JQdvV9cX786FMCYeihgZbW.png)

参考:

[基于RTX3090的Ubuntu22.04安装cuda11.7和cudnn8.6_ubuntu虚拟环境安装cuda11.7-博客]( "基于RTX3090的Ubuntu22.04安装cuda11.7和cudnn8.6_ubuntu虚拟环境安装cuda11.7-博客")

[炼丹学习笔记2---ubuntu2004运行3D Gaussian Splatting(3DGS)分享_ubuntu gaussian splatting torch-博客]( "炼丹学习笔记2---ubuntu2004运行3D Gaussian Splatting(3DGS)分享_ubuntu gaussian splatting torch-博客")

感谢!!!

全部评论 (0)

还没有任何评论哟~