Use runfiles to facilitate installing multiple CUDA versions

• Use runfiles to allow multiple CUDA versions to be installed on the same host
• Install cuDNN libraries
• Install helper tool to switch between CUDA versions (updating PATH and LD_LIBRARY_PATH)

defaults/main.yml

@@ -19,3 +19,30 @@ cuda_supported_arch:
 use_nvidia_cuda_repo: False
 
 upgrade_cuda: False
+
+cuda_runfile_urls:
+  11-0:
+    url: "https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run"
+    checksum: "md5:70af4cebe30549b9995fb9c57d538214"
+    install_dir: "/usr/local/cuda-11.0/"
+  11-1:
+    url: "https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run"
+    checksum: "md5:c24e2755e3868692051a38797ce01044"
+    install_dir: "/usr/local/cuda-11.1/"
+
+cuda_runfile_download_dir: "/root/cuda/"
+
+cuda_runfile_dependencies:
+  - gcc
+
+cudnn_urls:
+  11-0:
+    url: http://mrepoman01.cf.ac.uk/cudnn/cudnn-11.0-linux-x64-v8.0.5.39.tgz
+    checksum: "sha256:4e16ee7895deb4a8b1c194b812ba49586ef7d26902051401d3717511898a9b73"
+  11-1:
+    url: http://mrepoman01.cf.ac.uk/cudnn/cudnn-11.2-linux-x64-v8.1.1.33.tgz
+    checksum: "sha256:98a8784e92862f20018d20c281b30d4a0cd951f93694f6433ccf4ae9c502ba6a"
+
+cuda_runfile_install_cudnn: False
+
+cudnn_download_dir: "{{ cuda_runfile_download_dir }}cudnn/"
\ No newline at end of file