Skip to content

Commit 3edd6e3

Browse files
authored
Update README.rst
install step updates
1 parent feb824c commit 3edd6e3

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

README.rst

+2-3
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ Option 1: Install from PIP (recommended for analysts):
4747

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

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

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

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

7271
Recommended Citation

0 commit comments

Comments
 (0)