torch==2.5.1 | |
torchvision --index-url https://download.pytorch.org/whl/cu121 | |
nvidia-pyindex | |
cuda-python | |
nvidia-cuda-nvcc-cu12 | |
#nvidia-pytools | |
ray[default] | |
rich | |
setuptools | |
flatten-dict | |
xformers | |
numpy<2 | |
ninja | |
opencv-python | |
einops | |
tyro | |
accelerate==0.28.0 | |
imageio | |
imageio-ffmpeg | |
lpips | |
Pillow | |
safetensors | |
scikit-image | |
scikit-learn | |
scipy | |
tqdm | |
kiui >= 0.2.3 | |
roma | |
plyfile | |
# mvdream | |
diffusers==0.27.2 | |
huggingface_hub==0.25.2 | |
transformers | |
gsplat |