add accelerate dependency

Summary: Accelerate is an additional implicitron dependency, so document it.

Reviewed By: shapovalov

Differential Revision: D37786933

fbshipit-source-id: 11024fe604107881f8ca29e17cb5cbfe492fc7f9
This commit is contained in:
Jeremy Reizenstein 2022-07-13 06:00:05 -07:00 committed by Facebook GitHub Bot
parent 4261e59f51
commit 55f67b0d18
2 changed files with 11 additions and 4 deletions

View File

@ -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.

View File

@ -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": [