build: only install cuda-nvprof on x86_64

Signed-off-by: Esteve Fernandez <[email protected]>
autowarefoundation · Nov 3, 2023 · f5275c5 · f5275c5
1 parent bc6a64e
commit f5275c5
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ansible/roles/cuda/tasks/main.yaml b/ansible/roles/cuda/tasks/main.yaml
@@ -34,13 +34,20 @@
       - cuda-minimal-build-{{ cuda__dash_case_cuda_version.stdout }}
       - cuda-libraries-dev-{{ cuda__dash_case_cuda_version.stdout }}
       - cuda-nvml-dev-{{ cuda__dash_case_cuda_version.stdout }}
-      - cuda-nvprof-{{ cuda__dash_case_cuda_version.stdout }}
       - libnpp-dev-{{ cuda__dash_case_cuda_version.stdout }}
       - libcusparse-dev-{{ cuda__dash_case_cuda_version.stdout }}
       - libcublas-dev-{{ cuda__dash_case_cuda_version.stdout }}
       - libnccl-dev
     update_cache: true
 
+- name: Install extra CUDA libraries for x86_64
+  become: true
+  ansible.builtin.apt:
+    name:
+      - cuda-nvprof-{{ cuda__dash_case_cuda_version.stdout }}
+    update_cache: true
+  when: cuda_architecture.stdout == "x86_64"
+
 - name: Install cuda-drivers
   become: true
   ansible.builtin.apt: