fix(cuda): install NVML development library (#4621)

Signed-off-by: ito-san <[email protected]>
autowarefoundation · Apr 16, 2024 · 038f8a6 · 038f8a6
1 parent 9866356
commit 038f8a6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ansible/roles/cuda/tasks/main.yaml b/ansible/roles/cuda/tasks/main.yaml
@@ -34,6 +34,7 @@
       - libcusparse-dev-{{ cuda__dash_case_cuda_version.stdout }}
       - libcublas-dev-{{ cuda__dash_case_cuda_version.stdout }}
       - libcurand-dev-{{ cuda__dash_case_cuda_version.stdout }}
+      - cuda-nvml-dev-{{ cuda__dash_case_cuda_version.stdout }}
     update_cache: true
   when: install_devel == 'true'