diff --git a/doc/install.md b/doc/install.md
index 05a0d3bdb0..783abf65be 100644
--- a/doc/install.md
+++ b/doc/install.md
@@ -61,7 +61,7 @@ We follow the virtual environment approach to install the tensorflow's Python in
 virtualenv -p python3 $tensorflow_venv
 source $tensorflow_venv/bin/activate
 pip install --upgrade pip
-pip install --upgrade tensorflow==2.3.0
+pip install --upgrade tensorflow
 ```
 It is highly recommanded to keep the consistency of the TensorFlow version for the python and C++ interfaces. 
 Everytime a new shell is started and one wants to use `DeePMD-kit`, the virtual environment should be activated by 
@@ -78,7 +78,7 @@ virtualenv -p python3.7 $tensorflow_venv
 ```
 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	
 ```bash	
-pip install --upgrade tensorflow-cpu==2.3.0	
+pip install --upgrade tensorflow-cpu
 ```
 To verify the installation, run
 ```bash