From 55cdba2bdf57d075a356a1e6ba2aa78570e3e051 Mon Sep 17 00:00:00 2001 From: Sheng Zha Date: Tue, 14 Jul 2020 18:21:33 -0700 Subject: [PATCH] update cu101 static build cuda version --- tools/setup_gpu_build_tools.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/setup_gpu_build_tools.sh b/tools/setup_gpu_build_tools.sh index e10d1f1570fb..0e6f9fbc8999 100755 --- a/tools/setup_gpu_build_tools.sh +++ b/tools/setup_gpu_build_tools.sh @@ -34,7 +34,7 @@ if [[ $VARIANT == cu102* ]]; then LIBCUDNN_VERSION='7.6.5.32-1+cuda10.2' LIBNCCL_VERSION='2.5.6-1+cuda10.2' elif [[ $VARIANT == cu101* ]]; then - CUDA_VERSION='10.1.105-1' + CUDA_VERSION='10.1.243-1' CUDA_PATCH_VERSION='10.1.0.105-1' LIBCUDA_VERSION='418.39-0ubuntu1' LIBCUDNN_VERSION='7.6.5.32-1+cuda10.1'