Skip to content

Commit bbcc221

Browse files
committed
[BUILD] Update DEFAULT_CUDA_VERSION and DEFAULT_CUDNN_VERSION in configure.py.
Signed-off-by: Tongxuan Liu <tongxuan.ltx@alibaba-inc.com>
1 parent 99e2804 commit bbcc221

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

configure.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,8 @@
3333
from distutils.spawn import find_executable as which
3434
# pylint: enable=g-import-not-at-top
3535

36-
_DEFAULT_CUDA_VERSION = '10'
37-
_DEFAULT_CUDNN_VERSION = '7'
36+
_DEFAULT_CUDA_VERSION = '11'
37+
_DEFAULT_CUDNN_VERSION = '8'
3838
_DEFAULT_TENSORRT_VERSION = '5'
3939
_DEFAULT_CUDA_COMPUTE_CAPABILITIES = '6.0,6.1,7.0,7.5,8.0'
4040

0 commit comments

Comments
 (0)