Skip to content
This repository has been archived by the owner on Feb 6, 2023. It is now read-only.

Commit

Permalink
cuDNN v8.2.2
Browse files Browse the repository at this point in the history
  • Loading branch information
takagi committed Jul 15, 2021
1 parent 1617d9c commit 594e305
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions docker.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,6 +60,7 @@
'cudnn80-cuda110', 'cudnn80-cuda111',
'cudnn81-cuda110', 'cudnn81-cuda111', 'cudnn81-cuda112',
'cudnn82-cuda110', 'cudnn82-cuda111', 'cudnn82-cuda112', 'cudnn82-cuda113',
'cudnn82-cuda114',
]
nccl_choices = [
'none',
Expand Down Expand Up @@ -105,6 +106,7 @@
'cuda111': ['cudnn80-cuda111', 'cudnn81-cuda111', 'cudnn82-cuda111'],
'cuda112': ['cudnn81-cuda112', 'cudnn82-cuda112'],
'cuda113': ['cudnn82-cuda113'],
'cuda114': ['cudnn82-cuda114'],
}
cuda_nccls = {
# CUDA 9 does not support nccl 1.3
Expand Down Expand Up @@ -688,13 +690,14 @@ def get_scipy_choices():
codes['cudnn81-cuda112'] = codes['cudnn81-cuda110']

codes['cudnn82-cuda110'] = cudnn_base.format(
cudnn='cudnn-11.3-linux-x64-v8.2.1.32',
cudnn_ver='v8.2.1',
sha256sum='39412acd9ef5dd27954b6b9f5df75bd381c5d7ceb7979af6c743a7f4521f9c77',
cudnn='cudnn-11.4-linux-x64-v8.2.2.26',
cudnn_ver='v8.2.2',
sha256sum='fbc631ce19688e87d7d2420403b20db97885b17f718f0f51d7e9fc0905d86e07',
)
codes['cudnn82-cuda111'] = codes['cudnn82-cuda110']
codes['cudnn82-cuda112'] = codes['cudnn82-cuda110']
codes['cudnn82-cuda113'] = codes['cudnn82-cuda110']
codes['cudnn82-cuda114'] = codes['cudnn82-cuda110']

# This is a test for CFLAGS and LDFLAGS to specify a directory where cuDNN is
# installed.
Expand Down

0 comments on commit 594e305

Please sign in to comment.