DiffusionSfM / requirements.txt
qitaoz's picture
Update requirements.txt
07c92f0 verified
raw
history blame
823 Bytes
# 1 β€” PyTorch & friends (official CUDA 12.1 wheels)
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.2.2+cu121
torchvision==0.17.2+cu121 # must match the Torch build
# 2 β€” Extra index that stores the PyTorch3D wheels
--extra-index-url https://miropsota.github.io/torch_packages_builder
# 3 β€” Pre-built PyTorch3D wheel β€” NO git line!
pytorch3d==0.7.7+pt2.2.2cu121
git+https://github.com/facebookresearch/pytorch3d.git@v0.7.7#egg=pytorch3d
black
einops
flake8
hydra-core
imageio
imageio-ffmpeg
ipdb
ipython
isort
jupyter
jupyter_black
jupyter_nbextensions_configurator
jupyterlab
kaleido
matplotlib
numpy==1.26.4
omegaconf
opencv-python
plotly
scikit-learn
scipy
timm
trimesh
tqdm
iopath
pandas
wandb
submitit
accelerate==0.32.1
gradio==4.44.1
pydantic==2.10.6
gdown
xformers==0.0.22