diff --git a/ansible/roles/cuda/tasks/main.yaml b/ansible/roles/cuda/tasks/main.yaml index 0a075de1cfb..4ecf84ec401 100644 --- 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'