gaussian-splatting icon indicating copy to clipboard operation
gaussian-splatting copied to clipboard

Ubuntu 24.04 & CUDA 12 Support

Open pens opened this issue 1 year ago • 21 comments
trafficstars

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.

simple-knn.patch.txt

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.

SIBR_viewers.patch.txt

CudaRasterizer

Also probably due to NVIDIA's headers, CudaRasterizer/CudaRasterizer/cuda_rasterizer/rasterizer_impl.h requires cstdint to be included.

pens avatar Aug 05 '24 05:08 pens