gaussian-splatting
gaussian-splatting copied to clipboard
Ubuntu 24.04 & CUDA 12 Support
Apologies as this isn't an issue per se. I just finished getting training and the local viewer working on Ubuntu 24.04 with CUDA 12. As the process was a little involved I hope that sharing the changes needed might be useful.
environment.yml
name: gaussian_splatting
channels:
- pytorch
- conda-forge
- defaults
- nvidia
dependencies:
- cuda-toolkit=12.4
- plyfile
- python=3.8
- pip=22.3.1
- pytorch
- torchaudio
- torchvision
- tqdm
- pytorch-cuda=12.4
- pip:
- submodules/diff-gaussian-rasterization
- submodules/simple-knn
submodules/simple-knn/simple-knn.cu
The newer NVIDIA headers (seem to) no longer export FLT_MIN & FLT_MAX, so float.h needs to be explicitly included.
SIBR_viewers
Both Embree and FFmpeg have removed parts of their APIs that SIBR was using. I didn't test this with the remote viewer at all, as I was really just trying to get the local viewer building and running.
I updated the FFmpeg part with something that seemed in the right direction, but it's probably broken. I really have no idea if I hit the Embree path or not.
CudaRasterizer
Also probably due to NVIDIA's headers, CudaRasterizer/CudaRasterizer/cuda_rasterizer/rasterizer_impl.h requires cstdint to be included.