This is an automated email from the ASF dual-hosted git repository.
lfrolov pushed a commit to branch DATALAB-2698
in repository https://gitbox.apache.org/repos/asf/incubator-datalab.git
The following commit(s) were added to refs/heads/DATALAB-2698 by this push:
new 0ea1fdc [DATALAB-2698]: fixed install_venv_pip_pkg usage
0ea1fdc is described below
commit 0ea1fdc4a3c08150a8106d721c1725e2245952f8
Author: leonidfrolov <[email protected]>
AuthorDate: Mon Feb 14 20:16:01 2022 +0200
[DATALAB-2698]: fixed install_venv_pip_pkg usage
---
.../src/general/lib/os/debian/notebook_lib.py | 6 +++---
1 file changed, 3 insertions(+), 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 3052753..fd6bc98 100644
--- a/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
+++ b/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
@@ -397,18 +397,18 @@ def install_tensor(os_user, cuda_version, cuda_file_name,
def ensure_venv_libs(os_user, libs):
if not exists(datalab.fab.conn, '/home/' + os_user +
'/.ensure_dir/venv_libs_ensured'):
- datalab.fab.conn.install_venv_pip_pkg(libs)
+ datalab.fab.install_venv_pip_pkg(libs)
datalab.fab.conn.sudo('touch /home/' + os_user +
'/.ensure_dir/venv_libs_ensured')
def ensure_pytorch(os_user, gpu=True):
if not exists(datalab.fab.conn, '/home/' + os_user +
'/.ensure_dir/pytorch_ensured'):
if gpu:
- datalab.fab.conn.install_venv_pip_pkg('torch==1.10.2+cu113
torchvision==0.11.3+cu113 '
+ datalab.fab.install_venv_pip_pkg('torch==1.10.2+cu113
torchvision==0.11.3+cu113 '
'torchaudio==0.10.2+cu113 -f
'
'https://download.pytorch.org/whl/cu113/torch_stable.html')
else:
- datalab.fab.conn.install_venv_pip_pkg('torch==1.10.2+cpu
torchvision==0.11.3+cpu torchaudio==0.10.2+cpu -f '
+ datalab.fab.install_venv_pip_pkg('torch==1.10.2+cpu
torchvision==0.11.3+cpu torchaudio==0.10.2+cpu -f '
'https://download.pytorch.org/whl/cpu/torch_stable.html')
datalab.fab.conn.sudo('touch /home/' + os_user +
'/.ensure_dir/pytorch_ensured')
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
