From 55f67b0d1840df33e86e138ef8ae515b1977b4b4 Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Wed, 13 Jul 2022 06:00:05 -0700 Subject: [PATCH] add accelerate dependency Summary: Accelerate is an additional implicitron dependency, so document it. Reviewed By: shapovalov Differential Revision: D37786933 fbshipit-source-id: 11024fe604107881f8ca29e17cb5cbfe492fc7f9 --- projects/implicitron_trainer/README.md | 4 ++-- setup.py | 11 +++++++++-- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/projects/implicitron_trainer/README.md b/projects/implicitron_trainer/README.md index 5376c422..7451415b 100644 --- a/projects/implicitron_trainer/README.md +++ b/projects/implicitron_trainer/README.md @@ -27,7 +27,7 @@ Only configuration can be changed (see [Configuration system](#configuration-sys For this setup, install the dependencies and PyTorch3D from conda following [the guide](https://github.com/facebookresearch/pytorch3d/blob/master/INSTALL.md#1-install-with-cuda-support-from-anaconda-cloud-on-linux-only). Then, install implicitron-specific dependencies: ```shell -pip install "hydra-core>=1.1" visdom lpips matplotlib +pip install "hydra-core>=1.1" visdom lpips matplotlib accelerate ``` Runner executable is available as `pytorch3d_implicitron_runner` shell command. @@ -49,7 +49,7 @@ Please follow the instructions to [install PyTorch3D from a local clone](https:/ Then, install Implicitron-specific dependencies: ```shell -pip install "hydra-core>=1.1" visdom lpips matplotlib +pip install "hydra-core>=1.1" visdom lpips matplotlib accelerate ``` You are still encouraged to implement custom plugins as above where possible as it makes reusing the code easier. diff --git a/setup.py b/setup.py index 6b3f6800..48223710 100755 --- a/setup.py +++ b/setup.py @@ -143,8 +143,15 @@ setup( install_requires=["fvcore", "iopath"], extras_require={ "all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"], - "dev": ["flake8", "usort", "black==22.3"], - "implicitron": ["hydra-core>=1.1", "visdom", "lpips", "matplotlib"], + "dev": ["flake8", "usort"], + "implicitron": [ + "hydra-core>=1.1", + "visdom", + "lpips", + "tqdm>4.29.0", + "matplotlib", + "accelerate", + ], }, entry_points={ "console_scripts": [