Commit 3f2b6d51 authored by Dan Wilcox's avatar Dan Wilcox
Browse files

added version as download_tensorflow script first argument

parent 7fbff5fb
......@@ -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
......
......@@ -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
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment