build: keep /usr/local/cuda as path

Signed-off-by: Esteve Fernandez <[email protected]>
autowarefoundation · Nov 2, 2023 · a35c308 · a35c308
1 parent 495e85b
commit a35c308
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -52,15 +52,15 @@
 - 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
 
 - 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