You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@datalab.apache.org by lf...@apache.org on 2021/01/26 12:13:18 UTC
[incubator-datalab] branch DATALAB-2091 updated: [DATALAB-2091]:
changed how tensoeflow-gpu is installed
This is an automated email from the ASF dual-hosted git repository.
lfrolov pushed a commit to branch DATALAB-2091
in repository https://gitbox.apache.org/repos/asf/incubator-datalab.git
The following commit(s) were added to refs/heads/DATALAB-2091 by this push:
new 03c9866 [DATALAB-2091]: changed how tensoeflow-gpu is installed
03c9866 is described below
commit 03c9866fbe9106a0bb006863736efbaa93a527d6
Author: leonidfrolov <fr...@gmail.com>
AuthorDate: Tue Jan 26 14:13:00 2021 +0200
[DATALAB-2091]: changed how tensoeflow-gpu is installed
---
infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py b/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
index b44dd44..492d263 100644
--- a/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
+++ b/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
@@ -312,9 +312,7 @@ def install_tensor(os_user, cuda_version, cuda_file_name,
'echo "export LD_LIBRARY_PATH=\"$LD_LIBRARY_PATH:/opt/cudnn/lib64:/usr/local/cuda/lib64\"" >> ~/.bashrc')
# install TensorFlow and run TensorBoard
# sudo('python2.7 -m pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-{}-cp27-none-linux_x86_64.whl --no-cache-dir'.format(tensorflow_version))
- sudo('python3 -m pip install --upgrade '
- 'https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-{}-cp36-cp36m-manylinux2010_x86_64.whl'
- ' --no-cache-dir'.format(tensorflow_version))
+ sudo('python3 -m pip install --upgrade tensorflow-gpu=={} --no-cache-dir'.format(tensorflow_version))
sudo('mkdir /var/log/tensorboard; chown {0}:{0} -R /var/log/tensorboard'.format(os_user))
put('{}tensorboard.service'.format(templates_dir), '/tmp/tensorboard.service')
sudo("sed -i 's|OS_USR|{}|' /tmp/tensorboard.service".format(os_user))
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@datalab.apache.org
For additional commands, e-mail: commits-help@datalab.apache.org