added version as download_tensorflow script first argument

README.md

@@ -122,6 +122,12 @@ When opting for GPU support set the `TYPE` script variable:
 TYPE=gpu ./scripts/download_tensorflow.sh
 ```
 
+Additionally, to use a specific version supply it as the first argument:
+
+```shell
+./scripts/download_tensorflow.sh 2.7.0
+```
+
 See <https://www.tensorflow.org/install/gpu> for more information on GPU support for TensorFlow.
 
 ### Ubuntu / Linux

scripts/download_tensorflow.sh

@@ -8,8 +8,11 @@
 # stop on error
 set -e
 
-# tf version
+# tf version: optional argument
 VER=2.7.0
+if [ "$1" != "" ] ; then
+	VER=$1
+fi
 
 # tf type: cpu or gpu,
 # override when running via: TYPE=gpu ./download_tensorflow.sh