Enable user configurable cuda version

CICD/vars/versions.yml

@@ -31,3 +31,5 @@ nvidia_mig_parted_src_dir: /opt/src/mig-parted-0.1.3
 nvidia_mig_slurm_discovery_version: master
 nvidia_mig_slurm_discovery_src_url: https://gitlab.com/nvidia/hpc/slurm-mig-discovery.git
 nvidia_mig_slurm_discovery_src_dir: /opt/src/mig-slurm_discovery
+
+nvidia_cuda_version: cuda

roles/nvidia_mig_tools/tasks/main.yml

@@ -25,7 +25,7 @@
       become: true
     - name: install cuda
       apt:
-        name: cuda
+        name: {{ nvidia_cuda_version }}
         state: present
         update_cache: yes
       become: true