Skip to content

Commit 86cb0aa

Browse files
author
Taylor Robie
authored
Add requirements.txt to official. (#3760)
* add requirements.txt now that there are dependencies beyond tensorflow * direct pip info to README
1 parent 31fc896 commit 86cb0aa

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

official/README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,6 @@ If you would like to make any fixes or improvements to the models, please [submi
2828

2929
The *Official Models* are made available as a Python module. To run the models and associated scripts, add the top-level ***/models*** folder to the Python path with the command: `export PYTHONPATH="$PYTHONPATH:/path/to/models"`
3030

31+
To install dependencies pass `-r official/requirements.txt` to pip. (i.e. `pip3 install --user -r official/requirements.txt`)
32+
3133
To make Official Models easier to use, we are planning to create a pip installable Official Models package. This is being tracked in [#917](https://github.com/tensorflow/models/issues/917).

official/requirements.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
psutil>=5.4.3
2+
py-cpuinfo>=3.3.0

official/utils/logging/logger.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,7 @@
1616
"""Logging utilities for benchmark.
1717
1818
For collecting local environment metrics like CPU and memory, certain python
19-
packages need be installed. Run the following commands for dependency packages:
20-
> pip install --upgrade py-cpuinfo
21-
> pip install --upgrade psutil
19+
packages need be installed. See README for details.
2220
"""
2321
from __future__ import absolute_import
2422
from __future__ import division

0 commit comments

Comments
 (0)