File tree 1 file changed +2
-3
lines changed
1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ Option 1: Install from PIP (recommended for analysts):
47
47
48
48
3. Install sup3r: ``pip install NREL-sup3r ``
49
49
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] ``
51
51
52
52
4.1 For OSX use instead: ``python -m pip install tensorflow-metal ``
53
53
@@ -65,8 +65,7 @@ Option 2: Clone repo (recommended for developers)
65
65
5) Install ``sup3r `` and its dependencies by running:
66
66
``pip install . `` (or ``pip install -e . `` if running a dev branch
67
67
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] ``
70
69
7) *Optional *: Set up the pre-commit hooks with ``pip install pre-commit `` and ``pre-commit install ``
71
70
72
71
Recommended Citation
You can’t perform that action at this time.
0 commit comments