提交 f817465d 编写于 作者: H HexToString

Test Paddle2.1

上级 6d78d243
......@@ -34,13 +34,13 @@ message( "WITH_GPU = ${WITH_GPU}")
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")
......
......@@ -9,6 +9,11 @@ grpcio>=1.28.1
func-timeout>=4.3.5
pyyaml>=1.3.0
flask>=1.1.2
click==7.1.2
itsdangerous==1.1.0
Jinja2==2.11.3
MarkupSafe==1.1.1
Werkzeug==1.0.1
ujson>=2.0.3
sentencepiece==0.1.92; platform_machine != "aarch64"
sentencepiece; platform_machine == "aarch64"
......
......@@ -8,6 +8,11 @@ protobuf>=3.12.2
func-timeout>=4.3.5
pyyaml>=1.3.0
flask>=1.1.2
click==7.1.2
itsdangerous==1.1.0
Jinja2==2.11.3
MarkupSafe==1.1.1
Werkzeug==1.0.1
ujson>=2.0.3
grpcio-tools>=1.33.2
grpcio>=1.33.2
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册