From f5275c56aa6baeb5da336923d54f3c5edf739da6 Mon Sep 17 00:00:00 2001 From: Esteve Fernandez Date: Fri, 3 Nov 2023 16:35:02 +0100 Subject: [PATCH] build: only install cuda-nvprof on x86_64 Signed-off-by: Esteve Fernandez --- ansible/roles/cuda/tasks/main.yaml | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/ansible/roles/cuda/tasks/main.yaml b/ansible/roles/cuda/tasks/main.yaml index 0fecebba99f..81dd8ada902 100644 --- 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: