Skip to content

Commit

Permalink
Update README.rst
Browse files Browse the repository at this point in the history
install step updates
  • Loading branch information
bnb32 authored Nov 5, 2024
1 parent feb824c commit 3edd6e3
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions README.rst
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ Option 1: Install from PIP (recommended for analysts):

3. Install sup3r: ``pip install NREL-sup3r``

4. Run this if you want to train models on GPUs: ``conda install -c anaconda tensorflow-gpu``
4. Run this if you want to train models on GPUs: ``pip install tensorflow[and-cuda]``

4.1 For OSX use instead: ``python -m pip install tensorflow-metal``

Expand All @@ -65,8 +65,7 @@ Option 2: Clone repo (recommended for developers)
5) Install ``sup3r`` and its dependencies by running:
``pip install .`` (or ``pip install -e .`` if running a dev branch
or working on the source code)
6) Run this if you want to train models on GPUs: ``conda install -c anaconda tensorflow-gpu``
On Eagle HPC, you will need to also run ``pip install protobuf==3.20.*`` and ``pip install chardet``
6) Run this if you want to train models on GPUs: ``pip install tensorflow[and-cuda]``
7) *Optional*: Set up the pre-commit hooks with ``pip install pre-commit`` and ``pre-commit install``

Recommended Citation
Expand Down

0 comments on commit 3edd6e3

Please sign in to comment.