423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-03 20:32:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

add accelerate dependency

Browse Source 
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
projects/implicitron_trainer/README.md
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: 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 ```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. 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: Then, install Implicitron-specific dependencies:
```shell ```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. You are still encouraged to implement custom plugins as above where possible as it makes reusing the code easier.

11
setup.py
View File

@ -143,8 +143,15 @@ setup(
install_requires=["fvcore", "iopath"], install_requires=["fvcore", "iopath"],
extras_require={ extras_require={
"all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"], "all": ["matplotlib", "tqdm>4.29.0", "imageio", "ipywidgets"],
"dev": ["flake8", "usort", "black==22.3"], "dev": ["flake8", "usort"],
"implicitron": ["hydra-core>=1.1", "visdom", "lpips", "matplotlib"], "implicitron": [
"hydra-core>=1.1",
"visdom",
"lpips",
"tqdm>4.29.0",
"matplotlib",
"accelerate",
],
}, },
entry_points={ entry_points={
"console_scripts": [ "console_scripts": [