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/03/18 11:55:15 UTC

[incubator-datalab] 04/07: [DATALAB-2091]: fixed cuda installation directory path

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

commit 742b5d5a6d51a8152781206b73ee94ae4a33aa82
Author: leonidfrolov <fr...@gmail.com>
AuthorDate: Wed Mar 17 17:00:02 2021 +0200

    [DATALAB-2091]: fixed cuda installation directory path
---
 .../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 ad6c3cd..3834bae 100644
--- a/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
+++ b/infrastructure-provisioning/src/general/lib/os/debian/notebook_lib.py
@@ -298,9 +298,9 @@ def install_tensor(os_user, cuda_version, cuda_file_name,
             datalab.fab.conn.sudo('wget -P /opt https://developer.download.nvidia.com/compute/cuda/{0}/local_installers/{1}'.format(
                 cuda_version, cuda_file_name))
             datalab.fab.conn.sudo('sh /opt/{} --silent --toolkit'.format(cuda_file_name))
-            datalab.fab.conn.sudo('mv /usr/local/cuda-{} /opt/'.format(cuda_version))
-            datalab.fab.conn.sudo('ln -s /opt/cuda-{0} /usr/local/cuda-{0}'.format(cuda_version))
-            datalab.fab.conn.sudo('rm -f /opt/{}'.format(cuda_file_name))
+            datalab.fab.conn.sudo('mv /usr/local/cuda-{} /opt/'.format(cuda_version[:-2])) # for cuda 11.2.2 installation directory is 11.2
+            datalab.fab.conn.sudo('ln -s /opt/cuda-{0} /usr/local/cuda-{0}'.format(cuda_version[:-2])) # for cuda 11.2.2 installation directory is 11.2
+            datalab.fab.conn.sudo('rm -f /opt/{}'.format(cuda_file_name[:-2])) # for cuda 11.2.2 installation directory is 11.2
             # install cuDNN
             datalab.fab.conn.run('wget http://developer.download.nvidia.com/compute/redist/cudnn/v{0}/{1} -O /tmp/{1}'.format(
                 cudnn_version, cudnn_file_name))


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