diff --git a/cmake/paddlepaddle.cmake b/cmake/paddlepaddle.cmake index aac2791b4c16ab275d78d98ccb9abcb3d2eaa33c..78c398010153d9aa482388bfad34cb71ffd01c6e 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")