diff --git a/defaults/main.yml b/defaults/main.yml index 9cdf02664e5472bb65d27a2e468035d5343339dd..7f758133fc1b556a5f7b689c58d63330cbb82232 100644 --- a/defaults/main.yml +++ b/defaults/main.yml @@ -29,6 +29,10 @@ cuda_runfile_urls: 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/" + 11-2: + url: "https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run" + checksum: "md5:955ae5d59531923d32a3bfe3f68a7dfe" + install_dir: "/usr/local/cuda-11.2/" cuda_runfile_download_dir: "/root/cuda/" @@ -42,7 +46,10 @@ cudnn_urls: 11-1: url: http://mrepoman01.cf.ac.uk/cudnn/cudnn-11.2-linux-x64-v8.1.1.33.tgz checksum: "sha256:98a8784e92862f20018d20c281b30d4a0cd951f93694f6433ccf4ae9c502ba6a" + 11-2: + 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 +cudnn_download_dir: "{{ cuda_runfile_download_dir }}cudnn/" diff --git a/files/cuda-10-0.sh b/files/cuda.sh similarity index 80% rename from files/cuda-10-0.sh rename to files/cuda.sh index a3f4f237e40a941c370440cf5c71e84601ac6c10..58e3d92c844f24f69867f776d674e8d8b2b32d22 100644 --- a/files/cuda-10-0.sh +++ b/files/cuda.sh @@ -1,4 +1,4 @@ -for d in cuda-10.1 cuda-10.0 cuda; do +for d in cuda-11.2 cuda-11.1 cuda-10.1 cuda-10.0 cuda; do if [ -d "/usr/local/$d" ]; then CUDADIR="/usr/local/$d" break diff --git a/tasks/main.yml b/tasks/main.yml index b4d27a9cd74d27203d0ba7d2085b5da0a19df199..57a622285b0502fea052cf439b1a7bd64cd14ad9 100644 --- a/tasks/main.yml +++ b/tasks/main.yml @@ -152,6 +152,7 @@ loop: "{{ cuda_runfile_versions }}" when: cuda_runfile_versions is defined register: runfiles_downloaded + ignore_errors: "{{ ansible_check_mode }}" tags: - cuda-install @@ -181,6 +182,7 @@ loop: "{{ cuda_runfile_versions }}" when: (cuda_runfile_versions is defined) and cuda_runfile_install_cudnn register: cudnn_downloaded + ignore_errors: "{{ ansible_check_mode }}" tags: - cuda-install - cuda-cudnn @@ -213,9 +215,16 @@ - cuda-install - cuda-cudnn +- name: remove legacy cuda profile script + file: + path: /etc/profile.d/cuda-10-0.sh + state: absent + tags: + - cuda-config + - name: configure shell profile copy: - src: cuda-10-0.sh + src: cuda.sh dest: /etc/profile.d/ owner: root group: root