From 5233b7fdfe6a3d5e808500a06559f41ed43a32c6 Mon Sep 17 00:00:00 2001 From: HexToString <506181616@qq.com> Date: Thu, 13 May 2021 07:19:18 +0000 Subject: [PATCH] test2.1.0 error --- cmake/paddlepaddle.cmake | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/paddlepaddle.cmake b/cmake/paddlepaddle.cmake index aac2791b..78c39801 100644 --- a/cmake/paddlepaddle.cmake +++ b/cmake/paddlepaddle.cmake @@ -31,16 +31,16 @@ message( "WITH_GPU = ${WITH_GPU}") # Paddle Version should be one of: # latest: latest develop build # version number like 1.5.2 -SET(PADDLE_VERSION "2.0.1") +SET(PADDLE_VERSION "2.1.0") if (WITH_GPU) if(CUDA_VERSION EQUAL 11.0) - set(CUDA_SUFFIX "cuda11-cudnn8-avx-mkl") + set(CUDA_SUFFIX "cuda11.0-cudnn8-mkl-gcc8.2") set(WITH_TRT ON) elseif(CUDA_VERSION EQUAL 10.2) - set(CUDA_SUFFIX "cuda10.2-cudnn8-avx-mkl") + set(CUDA_SUFFIX "cuda10.2-cudnn8-mkl-gcc8.2") set(WITH_TRT ON) elseif(CUDA_VERSION EQUAL 10.1) - set(CUDA_SUFFIX "cuda10.1-cudnn7-avx-mkl") + set(CUDA_SUFFIX "cuda10.1-cudnn7-mkl-gcc8.2") set(WITH_TRT ON) elseif(CUDA_VERSION EQUAL 10.0) set(CUDA_SUFFIX "cuda10-cudnn7-avx-mkl") -- GitLab