You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@dlab.apache.org by lf...@apache.org on 2020/05/29 12:40:24 UTC
[incubator-dlab] 01/03: [DLAB-1810]: moved pytorch_branch as
variable to dlab.ini
This is an automated email from the ASF dual-hosted git repository.
lfrolov pushed a commit to branch DLAB-1810
in repository https://gitbox.apache.org/repos/asf/incubator-dlab.git
commit 825ffbf0d25d9d1d5b39ed2d6542d77dd305382b
Author: leonidfrolov <fr...@gmail.com>
AuthorDate: Fri May 29 10:40:43 2020 +0300
[DLAB-1810]: moved pytorch_branch as variable to dlab.ini
---
infrastructure-provisioning/src/general/conf/dlab.ini | 2 ++
infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py | 2 +-
2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/infrastructure-provisioning/src/general/conf/dlab.ini b/infrastructure-provisioning/src/general/conf/dlab.ini
index 5dbed5d..df7a781 100644
--- a/infrastructure-provisioning/src/general/conf/dlab.ini
+++ b/infrastructure-provisioning/src/general/conf/dlab.ini
@@ -252,6 +252,8 @@ nvidia_version = 418.126.02
caffe_version = 1.0
### Caffe2 library version for DeepLearning notebook
caffe2_version = 1.5
+### Pytorch branch used during caffe2 installation for DeepLearning notebook
+pytorch_branch = release/1.5
### Cmake version for DeepLearning notebook
cmake_version = 3.15.5
### CNTK library version for DeepLearning notebook
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 f36389c..f00d20f 100644
--- a/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
+++ b/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
@@ -460,7 +460,7 @@ def install_caffe2(os_user, caffe2_version, cmake_version):
with cd('/home/{}/pytorch/'.format(os_user)):
sudo('git submodule update --init')
with settings(warn_only=True):
- sudo('git checkout release/1.5')
+ sudo('git checkout {}'.format(os.environ['notebook_pytorch_branch']))
sudo('git submodule update --init --recursive')
sudo('python3 setup.py install')
sudo('touch /home/' + os_user + '/.ensure_dir/caffe2_ensured')
---------------------------------------------------------------------
To unsubscribe, e-mail: commits-unsubscribe@dlab.apache.org
For additional commands, e-mail: commits-help@dlab.apache.org