---
- name: "Add cuda-rhel7 repo for CUDA toolkit"
  ansible.builtin.command: yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
  args:
    creates: /etc/yum.repos.d/cuda-rhel7.repo
  notify: yum-clean-metadata


- name: "Install CUDA toolkit"
  ansible.builtin.yum:
    name:
      - "cuda-toolkit-{{ cuda_version }}"
      - "{{ cudnn_version }}"
      - "{{ cudnn_version }}-devel"

- name: Adding CUDA path
  ansible.builtin.blockinfile:
    path: /etc/profile.d/cuda.sh
    create: yes
    block: |
      export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
      export PATH=$PATH:/usr/local/cuda/bin
      export CPATH=${CPATH}:/usr/local/cuda/targets/x86_64-linux/include

# - name: clock
#   ansible.builtin.blockinfile:
#     path: /etc/profile.d/cuda-tune.sh
#     create: yes
#     block: |
#       nvidia-smi -pm 1