summaryrefslogtreecommitdiffstats
path: root/roles/cuda
diff options
context:
space:
mode:
Diffstat (limited to 'roles/cuda')
-rw-r--r--roles/cuda/defaults/main.yml6
-rw-r--r--roles/cuda/tasks/cuda_init.yml5
-rw-r--r--roles/cuda/tasks/main.yml10
-rw-r--r--roles/cuda/vars/fedora-28.yml4
4 files changed, 17 insertions, 8 deletions
diff --git a/roles/cuda/defaults/main.yml b/roles/cuda/defaults/main.yml
index 57efd6b..45ceb29 100644
--- a/roles/cuda/defaults/main.yml
+++ b/roles/cuda/defaults/main.yml
@@ -2,14 +2,14 @@
# defaults file for ansible-role-cuda
# By default gpu is False, set it to True on the gpu nodes in the hosts file
-gpu: False
+gpu: True
cuda_repo_url: "http://developer.download.nvidia.com/compute/cuda/repos/"
cuda_rpm_key_path: /etc/rpm/nvidia_packaging_key.asc
cuda_packages:
- cuda
-cuda_restart_node_on_install: True
+cuda_restart_node_on_install: False
cuda_init: True
-cuda_init_restart_service: True
+cuda_init_restart_service: False
cuda_init_compute_mode: 3
cuda_init_persistence_mode: 1
cuda_gpu_name0: "/dev/nvidia0"
diff --git a/roles/cuda/tasks/cuda_init.yml b/roles/cuda/tasks/cuda_init.yml
index 9a10284..ff54994 100644
--- a/roles/cuda/tasks/cuda_init.yml
+++ b/roles/cuda/tasks/cuda_init.yml
@@ -36,10 +36,7 @@
path: "{{ cuda_gpu_name0 }}"
register: reg_cuda_gpu_name0
check_mode: no
-
-- debug:
- var: reg_cuda_gpu_name0
- verbosity: 1
+ failed_when: false
- name: Initialize the GPUs - run cuda_init.sh if there is no /dev/nvidia0
command: /bin/bash /usr/local/bin/cuda_init.sh
diff --git a/roles/cuda/tasks/main.yml b/roles/cuda/tasks/main.yml
index 6d846f8..f292f67 100644
--- a/roles/cuda/tasks/main.yml
+++ b/roles/cuda/tasks/main.yml
@@ -10,11 +10,19 @@
- block:
- include_tasks: configure_yum.yml
- when: ansible_pkg_mgr == 'yum'
+ when: ansible_pkg_mgr == 'yum' or ansible_pkg_mgr == 'dnf'
- include_tasks: configure_apt.yml
when: ansible_pkg_mgr == 'apt'
+ - name: Install kernel development files
+ package: name=kernel-devel state=present
+ register: result
+
+ - name: Synchronize kernel and kernel-devel packages
+ package: name=kernel state=latest
+ when: (result | changed)
+
- name: Install CUDA and related packages (1.5-2GB download, also restarts if cuda_restart_node_on_install is set to True)
package:
name: "{{ item }}"
diff --git a/roles/cuda/vars/fedora-28.yml b/roles/cuda/vars/fedora-28.yml
new file mode 100644
index 0000000..e8af2d0
--- /dev/null
+++ b/roles/cuda/vars/fedora-28.yml
@@ -0,0 +1,4 @@
+---
+cuda_repo_subfolder: fedora27
+
+# vim:ft=ansible: \ No newline at end of file