This is an automated email from the ASF dual-hosted git repository.
lfrolov pushed a commit to branch develop
in repository https://gitbox.apache.org/repos/asf/incubator-datalab.git
The following commit(s) were added to refs/heads/develop by this push:
new 8b1fb2a [DATALAB-2708]: changed how tflite-runtime is installed
8b1fb2a is described below
commit 8b1fb2aec8339c5fd695527714c8d12879da5ee3
Author: leonidfrolov <[email protected]>
AuthorDate: Thu Feb 17 13:56:11 2022 +0200
[DATALAB-2708]: changed how tflite-runtime is installed
---
.../src/tensor/scripts/configure_tensor_node.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git
a/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
b/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
index 973ff59..1f78787 100644
--- a/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
+++ b/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
@@ -69,7 +69,7 @@ cuda_version = os.environ['notebook_cuda_version']
cuda_file_name = os.environ['notebook_cuda_file_name']
cudnn_version = os.environ['notebook_cudnn_version']
cudnn_file_name = os.environ['notebook_cudnn_file_name']
-venv_libs = 'numpy scipy matplotlib pandas scikit-learn opencv-python
tensorflow==2.5.0 tflite==2.5.0'
+venv_libs = 'numpy scipy matplotlib pandas scikit-learn opencv-python
tensorflow==2.5.0'
##############
# Run script #
@@ -150,6 +150,7 @@ if __name__ == "__main__":
print("Install python venv required libs")
ensure_venv_libs(args.os_user, venv_libs)
+ ensure_venv_libs(args.os_user, '--extra-index-url
https://google-coral.github.io/py-repo/ tflite_runtime==2.5.0')
#POST INSTALLATION PROCESS
print("Updating pyOpenSSL library")
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
