pgnd / requirements.txt
kaifz's picture
update
5205854
# 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
diff-gaussian-rasterization @ https://huggingface.co/spaces/kaifz/pgnd/resolve/main/wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl