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'