Support cuDNN 8.2.4

chainer · Sep 9, 2021 · 24031e1 · 24031e1
1 parent 23feaf8
commit 24031e1
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/docker.py b/docker.py
@@ -550,7 +550,7 @@ def get_scipy_choices():
 
 cudnn_base = '''
 WORKDIR /opt/cudnn
-RUN curl -s -o {cudnn}.tgz http://developer.download.nvidia.com/compute/redist/cudnn/{cudnn_ver}/{cudnn}.tgz && \\
+RUN curl -s -o {cudnn}.tgz https://developer.download.nvidia.com/compute/redist/cudnn/{cudnn_ver}/{cudnn}.tgz && \\
     echo "{sha256sum}  {cudnn}.tgz" | sha256sum -cw --quiet - && \\
     tar -xzf {cudnn}.tgz -C /usr/local && \\
     rm {cudnn}.tgz
@@ -689,9 +689,9 @@ def get_scipy_choices():
 codes['cudnn81-cuda112'] = codes['cudnn81-cuda110']
 
 codes['cudnn82-cuda110'] = cudnn_base.format(
-    cudnn='cudnn-11.4-linux-x64-v8.2.2.26',
-    cudnn_ver='v8.2.2',
-    sha256sum='fbc631ce19688e87d7d2420403b20db97885b17f718f0f51d7e9fc0905d86e07',
+    cudnn='cudnn-11.4-linux-x64-v8.2.4.15',
+    cudnn_ver='v8.2.4',
+    sha256sum='0e5d2df890b9967efa6619da421310d97323565a79f05a1a8cb9b7165baad0d7',
 )
 codes['cudnn82-cuda111'] = codes['cudnn82-cuda110']
 codes['cudnn82-cuda112'] = codes['cudnn82-cuda110']
@@ -701,7 +701,7 @@ def get_scipy_choices():
 # installed.
 codes['cudnn-latest-with-dummy'] = '''
 WORKDIR /opt/cudnn
-RUN curl -s -o {cudnn}.tgz http://developer.download.nvidia.com/compute/redist/cudnn/{cudnn_ver}/{cudnn}.tgz && \\
+RUN curl -s -o {cudnn}.tgz https://developer.download.nvidia.com/compute/redist/cudnn/{cudnn_ver}/{cudnn}.tgz && \\
     echo "{sha256sum}  {cudnn}.tgz" | sha256sum -cw --quiet - && \\
     tar -xzf {cudnn}.tgz -C /opt/cudnn && \\
     rm {cudnn}.tgz