DiffusionSfM / requirements.txt
qitaoz's picture
Update requirements.txt
367570f verified
raw
history blame contribute delete
807 Bytes
# ---------- PyTorch wheels that bundle the CUDA-12.1 runtime ----------
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.2.2+cu121
torchvision==0.17.2+cu121
# torch==2.4.0
# torchvision==0.19.0
# ---------- Extra index that hosts the PyTorch3D binary wheels ----------
--extra-index-url https://miropsota.github.io/torch_packages_builder
pytorch3d==0.7.8+pt2.2.2cu121
# pytorch3d==0.7.7+pt2.4.0cu121
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.25.post1
# xformers==0.0.27.post1