[incubator-datalab] branch develop updated: [DATALAB-2992]: Added pre-installation of Pytorch library

[lf...@apache.org]

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 a7eb47692 [DATALAB-2992]: Added pre-installation of Pytorch library
     new 72acb7a1b Merge pull request #1711 from OleksandrRepnikov/DATALAB-2992
a7eb47692 is described below

commit a7eb476925390f519de51eaa80815179f14baf9b
Author: orepnikov <60...@users.noreply.github.com>
AuthorDate: Mon Sep 5 12:33:21 2022 +0300

    [DATALAB-2992]: Added pre-installation of Pytorch library
---
 .../src/tensor/scripts/configure_tensor_node.py                      | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py b/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
index b1bef7b47..021c40060 100644
--- a/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
+++ b/infrastructure-provisioning/src/tensor/scripts/configure_tensor_node.py
@@ -142,9 +142,8 @@ if __name__ == "__main__":
     # INSTALL OPTIONAL PACKAGES
     print("Installing additional Python packages")
     ensure_additional_python_libs(args.os_user)
-    if os.environ['conf_cloud_provider'] == 'aws':
-        print('Installing Pytorch')
-        ensure_pytorch(args.os_user)
+    print('Installing Pytorch')
+    ensure_pytorch(args.os_user)
     print("Install Matplotlib")
     ensure_matplot(args.os_user)
 


---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@datalab.apache.org
For additional commands, e-mail: commits-help@datalab.apache.org