diff --git a/ansible/roles/cuda/README.md b/ansible/roles/cuda/README.md index 59c4ac71410..b5075576055 100644 --- a/ansible/roles/cuda/README.md +++ b/ansible/roles/cuda/README.md @@ -33,6 +33,6 @@ Perform the post installation actions: ```bash # Taken from: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions -echo 'export PATH=/usr/local/cuda-12.3/bin${PATH:+:${PATH}}' >> ~/.bashrc -echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc +echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrc +echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc ``` diff --git a/ansible/roles/cuda/tasks/main.yaml b/ansible/roles/cuda/tasks/main.yaml index c1c01d038e7..0fecebba99f 100644 --- a/ansible/roles/cuda/tasks/main.yaml +++ b/ansible/roles/cuda/tasks/main.yaml @@ -52,7 +52,7 @@ - name: Add PATH to .bashrc ansible.builtin.lineinfile: dest: ~/.bashrc - line: export PATH="/usr/local/cuda-12.3/bin:$PATH" + line: export PATH="/usr/local/cuda/bin:$PATH" state: present create: true mode: 0644 @@ -60,7 +60,7 @@ - name: Add LD_LIBRARY_PATH to bashrc ansible.builtin.lineinfile: dest: ~/.bashrc - line: export LD_LIBRARY_PATH="/usr/local/cuda-12.3/lib64:$LD_LIBRARY_PATH" + line: export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH" state: present create: true mode: 0644