bump to CUDA 10.2

chainer · Sep 22, 2021 · b6319ca · b6319ca
1 parent 5836ea8
commit b6319ca
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/run_test.py b/run_test.py
@@ -248,9 +248,9 @@ def main():
             requires = ['optuna']
         conf = {
             'base': 'ubuntu18_py39-pyenv',
-            'cuda': 'cuda100',
-            'cudnn': 'cudnn76-cuda100',
-            'nccl': 'nccl2.4-cuda100',
+            'cuda': 'cuda102',
+            'cudnn': 'cudnn76-cuda102',
+            'nccl': 'nccl2.5-cuda102',
             'cutensor': 'none',
             'cusparselt': 'none',
             'requires': [
@@ -280,9 +280,9 @@ def main():
     elif args.test == 'cupy-py3-cub':
         conf = {
             'base': 'ubuntu18_py38-pyenv',
-            'cuda': 'cuda100',
-            'cudnn': 'cudnn76-cuda100',
-            'nccl': 'nccl2.4-cuda100',
+            'cuda': 'cuda102',
+            'cudnn': 'cudnn76-cuda102',
+            'nccl': 'nccl2.5-cuda100',
             'cutensor': 'none',
             'cusparselt': 'none',
             'requires': [
@@ -386,9 +386,9 @@ def main():
         # NumPy 1.14.
         conf = {
             'base': 'ubuntu18_py38-pyenv',
-            'cuda': 'cuda100',
-            'cudnn': 'cudnn76-cuda100',
-            'nccl': 'nccl2.4-cuda100',
+            'cuda': 'cuda102',
+            'cudnn': 'cudnn76-cuda102',
+            'nccl': 'nccl2.5-cuda100',
             'cutensor': 'none',
             'cusparselt': 'none',
             'requires': [