From ecb3b759700f01c45ee38a7d62189d4f868ca695 Mon Sep 17 00:00:00 2001 From: "mergify[bot]" <37929162+mergify[bot]@users.noreply.github.com> Date: Thu, 18 Feb 2021 20:40:04 +0000 Subject: [PATCH] Merge pull request #4680 from emcastillo/cudnn-11.2 Add cudnn and cutensor for CUDA 11.2 in `install_library` --- cupyx/tools/install_library.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cupyx/tools/install_library.py b/cupyx/tools/install_library.py index 6f876bd1311..9cb5785965c 100755 --- a/cupyx/tools/install_library.py +++ b/cupyx/tools/install_library.py @@ -49,6 +49,10 @@ def _make_cudnn_record( # Latest cuDNN versions: https://developer.nvidia.com/rdp/cudnn-download +_cudnn_records.append(_make_cudnn_record( + '11.2', '8.1.0', + 'cudnn-11.2-linux-x64-v8.1.0.77.tgz', + 'cudnn-11.2-windows-x64-v8.1.0.77.zip')) _cudnn_records.append(_make_cudnn_record( '11.1', '8.1.0', 'cudnn-11.2-linux-x64-v8.1.0.77.tgz',