Ansible Role: cuDNN
Install cuDNN on Linux System.
Requirements
- CUDA Toolkit installed.
- Download cuDNN to
filesfolder with your NVIDIA Developer account.
Role Variables
cuda_version: The version of CUDA Toolkit installed. (default: 8.0)
cudnn_version: The version cuDNN. (default: v5.1)
Dependencies
None.
Example Playbook
- hosts: gpu
become: yes
roles:
- { role: cudnn, cuda_version: 8.0, cudnn_version: v5.1}
License
BSD
Author Information
This role was created by Daniel D.