GaussianPRO

pixi.toml

[project]
name = "GaussianPro"
version = "0.1.0"
description = "Add a short description here"
authors = ["pablovela5620 <pablovela5620@gmail.com>"]
channels = ["nvidia/label/cuda-11.8.0", "nvidia", "conda-forge", "pytorch"]
platforms = ["linux-64"]

[tasks]
torch-gpu-check = """
    python -c "import torch; print(torch.__version__);print(torch.cuda.is_available())"
"""
build = {cmd = "cmake . && make -j$(nproc)", cwd = "submodules/Propagation"}
diff-gauss = "python -m pip install submodules/diff-gaussian-rasterization"
simple-knn = "python -m pip install submodules/simple-knn"
post-install = {cmd = "pwd", depends_on=["build", "diff-gauss", "simple-knn"]}

[dependencies]
# Build tools:
clang-tools = ">=15,<16" # clang-format
cmake = "3.27.6"
# Required by this example:
eigen = "3.4.0.*"
opencv = "4.8.1.*"
python = "3.11.*"
pip = ">=23.3.2,<23.4"
cuda = {version = "*", channel="nvidia/label/cuda-11.8.0"}
pytorch-cuda = {version = "11.8.*", channel="pytorch"}
pytorch = {version = ">=2.2.0,<2.3", channel="pytorch"}
torchvision = {version = ">=0.17.0,<0.18", channel="pytorch"}
gdown = ">=5.1.0,<5.2"
tqdm = ">=4.66.2,<4.67"
plyfile = ">=1.0.3,<1.1"
matplotlib = ">=3.8.3,<3.9"
open3d = ">=0.18.0,<0.19"
imageio = ">=2.34.0,<2.35"