From 76f557901b916831dc906209d84d05660e473ed6 Mon Sep 17 00:00:00 2001 From: HexToString <506181616@qq.com> Date: Thu, 13 May 2021 04:53:11 +0000 Subject: [PATCH] test 201 --- cmake/paddlepaddle.cmake | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/cmake/paddlepaddle.cmake b/cmake/paddlepaddle.cmake index 78c39801..aac2791b 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.1.0") +SET(PADDLE_VERSION "2.0.1") if (WITH_GPU) if(CUDA_VERSION EQUAL 11.0) - set(CUDA_SUFFIX "cuda11.0-cudnn8-mkl-gcc8.2") + set(CUDA_SUFFIX "cuda11-cudnn8-avx-mkl") set(WITH_TRT ON) elseif(CUDA_VERSION EQUAL 10.2) - set(CUDA_SUFFIX "cuda10.2-cudnn8-mkl-gcc8.2") + set(CUDA_SUFFIX "cuda10.2-cudnn8-avx-mkl") set(WITH_TRT ON) elseif(CUDA_VERSION EQUAL 10.1) - set(CUDA_SUFFIX "cuda10.1-cudnn7-mkl-gcc8.2") + set(CUDA_SUFFIX "cuda10.1-cudnn7-avx-mkl") set(WITH_TRT ON) elseif(CUDA_VERSION EQUAL 10.0) set(CUDA_SUFFIX "cuda10-cudnn7-avx-mkl") -- GitLab