From b6319ca72150b39c43ca6e1c10803279bb462b5f Mon Sep 17 00:00:00 2001 From: Kenichi Maehashi Date: Wed, 22 Sep 2021 05:39:16 +0000 Subject: [PATCH] bump to CUDA 10.2 --- run_test.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/run_test.py b/run_test.py index b239f80..361b81d 100755 --- 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': [