pgnd / requirements.txt
kaifz's picture
update path
e7d809a
raw
history blame
1.44 kB
# warp-lang==1.1.0
trimesh
pyyaml
numpy==1.26.4
scipy
transformers
scikit-learn
scikit-image
pyvista
opencv-contrib-python
open3d
moviepy
matplotlib
kornia
ipdb
h5py
gradio
ffmpeg
einops
dill
av
timm
omegaconf
hydra-core
wandb
# torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
# torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
# torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
# diff-gaussian-rasterization @ file://./src/third-party/diff-gaussian-rasterization-w-depth
# dgl==2.4.0+cu124 --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
# --- core CUDA wheels -------------------------------------------------
--extra-index-url https://download.pytorch.org/whl/cu124
torch==2.4.0+cu124 # ⇦ install Torch first
torchvision==0.19.0+cu124
torchaudio==2.4.0+cu124
# --- DGL (needs the matching Torch 2.4 wheel) -------------------------
--find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
dgl==2.4.0+cu124
warp-lang==1.1.0
# ------------------------------------------------------------------
# Gaussian-splatting extension from GitHub
# (choose ONE of the two lines)
# 1) pin to the latest main commit
git+https://github.com/kywind/diff-gaussian-rasterization-w-depth.git
# 2) or pin to a release tag, e.g. v0.1.0
# git+https://github.com/kywind/[email protected]#egg=diff-gaussian-rasterization