3DGS代码中有两个通用的模块
–submodules
|----diff-gaussian-rasterization
|----simple-knn
安装不上这两个模块,网上跟着各个教程试了很多,没有能够走通的。
1. 如嫌麻烦可直接下载whl文件
链接: https://pan.baidu.com/s/1M5jeT1fMMrSS0xa9XJsWlg?pwd=bcfi 提取码: bcfi
放到工程文件夹下,然后pip install 该whl文件的路径即可, 如
pip install submodules/simple_knn-0.0.0-cp310-cp310-linux_x86_64.whl
安装要求
1、Python3.10
2、安装pytorch2.0.1 -cuda11.8
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
安装完毕,代码就可以正常运行了
2.下面为具体解决过程:
解决方式
前提
软件:ubuntu 20.04,cuda11.8,cmake 3.23.0,gcc 9.4.0
硬件:2060-12G,NTFS硬盘(报错,切勿使用)
注意:代码所在硬盘若为NTFS格式,在gitclone代码与pip install 时会出现权限问题,应当避免使用
解决过程
电脑安装cuda11.8环境
安装cuda11.8
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda
进入bashrc配置cuda环境
sudo vim ~/.bashrc
在bashrc文件最下方,添加下入代码,(ps:这边需要注意cuda的版本,版本不同,路径的命名需修改)
export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64
保存,更新环境
source ~/.bashrc
测试CUDA是否安装成功
nvcc -V
输出下述结果,表示安装成功:wit622@622:~/cuda$ nvcc -V
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0
参考链接:https://blog.csdn.net/w946612410/article/details/131786512
安装gcc 9.4.0
sudo apt-get update # 更新软件列表
sudo apt-get install g++ # 下载g++编译器
sudo apt-get install gcc # 下载gcc编译器
sudo apt-get install make # 下载GNU Make编译器
sudo apt-get install initramfs-tools # 下载安装initramfs-tools
安装cmake 3.23.0
wget https://cmake.org/files/v3.23/cmake-3.23.0.tar.gz
下载cmake文件后,进行安装
安装pytorch2.0.1 -cuda11.8
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
pip install submodules/diff-gaussian-rasterization 即可运行成功!!
题外:Python项目打包成wheel笔记
https://blog.csdn.net/yuan2019035055/article/details/128709587
安装setuptools
pip install setuptools
pip install wheel
安装微软的VS Install
pip install nes-py --no-cache-dir
pip install --upgrade pip setuptools wheel
打包为whl文件
python setup.py bdist_wheel
运行完成后会在setup.py同级目录下生成dist文件夹,里面的whl文件可以拷贝到其他电脑进行安装
python版本对应的其他.whl文件,下载地址:https://pypi.org/