mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-01 03:12:49 +08:00
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:
parent
4261e59f51
commit
55f67b0d18
@ -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.
|
||||
|
11
setup.py
11
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": [
|
||||
|
Loading…
x
Reference in New Issue
Block a user