This is an automated email from the ASF dual-hosted git repository.
mhladun pushed a commit to branch gcp-jupyter-highgpu-template-1
in repository https://gitbox.apache.org/repos/asf/incubator-datalab.git
The following commit(s) were added to refs/heads/gcp-jupyter-highgpu-template-1
by this push:
new 6405447 Gpu type parsing
new e988704 Merge branch 'gcp-jupyter-highgpu-template-1' of
https://github.com/apache/incubator-datalab into gcp-jupyter-highgpu-template-1
6405447 is described below
commit 640544720795a4a30ed4eb2c8bc02428de509e9e
Author: Marian_Hladun <[email protected]>
AuthorDate: Wed Mar 2 17:03:59 2022 +0200
Gpu type parsing
---
.../src/general/scripts/gcp/common_prepare_notebook.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git
a/infrastructure-provisioning/src/general/scripts/gcp/common_prepare_notebook.py
b/infrastructure-provisioning/src/general/scripts/gcp/common_prepare_notebook.py
index 814a227..a13d1bb 100644
---
a/infrastructure-provisioning/src/general/scripts/gcp/common_prepare_notebook.py
+++
b/infrastructure-provisioning/src/general/scripts/gcp/common_prepare_notebook.py
@@ -159,7 +159,12 @@ if __name__ == "__main__":
notebook_config['gpu_accelerator_type'] = 'None'
notebook_config['gpu_accelerator_count'] = 'None'
- if os.environ['application'] in ('tensor', 'tensor-rstudio',
'deeplearning') or os.environ['gpu_enabled'] == 'True':
+ # if os.environ['application'] == 'jupyter-gpu':
+ # notebook_config['gpu_accelerator_type'] == 'nvidia-tesla-a100'
+ # notebook_config['gpu_accelerator_count'] == '1'
+
+ if os.environ['application'] in ('tensor', 'tensor-rstudio',
'deeplearning', 'jupyter-gpu') or os.environ[
+ 'gpu_enabled'] == 'True':
if os.environ['gpuType'] != '':
notebook_config['gpu_accelerator_type'] = os.environ['gpuType']
notebook_config['gpu_accelerator_count'] =
os.environ['gpuCount']
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
