Skip to content

Commit d3f7cde

Browse files
authored
add document for cpu-only tensorflow
1 parent 1a81455 commit d3f7cde

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,10 @@ If one has multiple python interpreters named like python3.x, it can be specifie
135135
```bash
136136
virtualenv -p python3.7 $tensorflow_venv
137137
```
138+
If one does not need the GPU support of deepmd-kit and is concerned about package size, the CPU-only version of tensorflow should be installed by
139+
```bash
140+
pip install --upgrade tensorflow-cpu==2.1.0
141+
```
138142
To verify the installation, run
139143
```bash
140144
python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

0 commit comments

Comments
 (0)