Merge pull request #1010 from wangjiawei04/cmake

fix cmake and dockerfile

cmake/paddlepaddle.cmake

@@ -32,19 +32,23 @@ message( "WITH_GPU = ${WITH_GPU}")
 # latest: latest develop build
 # version number like 1.5.2
 SET(PADDLE_VERSION "2.0.0")
-if(CUDA_VERSION EQUAL 11.0)
-    set(CUDA_SUFFIX "cuda11-cudnn8-avx-mkl")
-    set(WITH_TRT ON)
-elseif(CUDA_VERSION EQUAL 10.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-avx-mkl")
-    set(WITH_TRT ON)
-elseif(CUDA_VERSION EQUAL 10.0)
-    set(CUDA_SUFFIX "cuda10-cudnn7-avx-mkl")
-elseif(CUDA_VERSION EQUAL 9.0)
-    set(CUDA_SUFFIX "cuda9-cudnn7-avx-mkl")
+if (WITH_GPU)
+    if(CUDA_VERSION EQUAL 11.0)
+        set(CUDA_SUFFIX "cuda11-cudnn8-avx-mkl")
+        set(WITH_TRT ON)
+    elseif(CUDA_VERSION EQUAL 10.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-avx-mkl")
+        set(WITH_TRT ON)
+    elseif(CUDA_VERSION EQUAL 10.0)
+        set(CUDA_SUFFIX "cuda10-cudnn7-avx-mkl")
+    elseif(CUDA_VERSION EQUAL 9.0)
+        set(CUDA_SUFFIX "cuda9-cudnn7-avx-mkl")
+    endif()
+else()
+    set(WITH_TRT OFF)
 endif()  
 
 if (WITH_GPU)

core/cube/cube-builder/CMakeLists.txt

@@ -22,7 +22,7 @@ include_directories(SYSTEM ${CMAKE_CURRENT_BINARY_DIR}/../)
 
 add_executable(cube-builder src/main.cpp include/cube-builder/util.h src/util.cpp src/builder_job.cpp include/cube-builder/builder_job.h include/cube-builder/define.h src/seqfile_reader.cpp include/cube-builder/seqfile_reader.h include/cube-builder/raw_reader.h include/cube-builder/vtext.h src/crovl_builder_increment.cpp include/cube-builder/crovl_builder_increment.h src/curl_simple.cpp include/cube-builder/curl_simple.h)
 
-add_dependencies(cube-builder jsoncpp)
+add_dependencies(cube-builder jsoncpp boost)
 
 set(DYNAMIC_LIB
     gflags

tools/Dockerfile.ci

-FROM centos:7.3.1611
+FROM hub.baidubce.com/paddlepaddle/serving:latest-cuda10.2-cudnn8-devel
 
-RUN yum -y install wget >/dev/null \
-    && yum -y install gcc gcc-c++ make glibc-static which >/dev/null \
-    && yum -y install git openssl-devel curl-devel bzip2-devel python-devel >/dev/null \
-    && yum -y install libSM-1.2.2-2.el7.x86_64 --setopt=protected_multilib=false \
-    && yum -y install libXrender-0.9.10-1.el7.x86_64 --setopt=protected_multilib=false \
-    && yum -y install libXext-1.3.3-3.el7.x86_64 --setopt=protected_multilib=false
-    
-RUN wget https://cmake.org/files/v3.2/cmake-3.2.0-Linux-x86_64.tar.gz >/dev/null \
-    && tar xzf cmake-3.2.0-Linux-x86_64.tar.gz \
-    && mv cmake-3.2.0-Linux-x86_64 /usr/local/cmake3.2.0 \
-    && echo 'export PATH=/usr/local/cmake3.2.0/bin:$PATH' >> /root/.bashrc \
-    && rm cmake-3.2.0-Linux-x86_64.tar.gz
-
-RUN wget https://dl.google.com/go/go1.14.linux-amd64.tar.gz >/dev/null \
-    && tar xzf go1.14.linux-amd64.tar.gz \
-    && mv go /usr/local/go \
-    && echo 'export GOROOT=/usr/local/go' >> /root/.bashrc \
-    && echo 'export PATH=/usr/local/go/bin:$PATH' >> /root/.bashrc \
-    && rm go1.14.linux-amd64.tar.gz
-    
-RUN yum -y install python-devel sqlite-devel >/dev/null \
-    && curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py >/dev/null \
-    && python get-pip.py >/dev/null \
-    && rm get-pip.py \
-    && pip install requests 
-
-RUN wget http://nixos.org/releases/patchelf/patchelf-0.10/patchelf-0.10.tar.bz2 \
-    && yum -y install bzip2 >/dev/null \
-    && tar -jxf patchelf-0.10.tar.bz2 \
-    && cd patchelf-0.10 \
-    && ./configure --prefix=/usr \
-    && make >/dev/null && make install >/dev/null \
-    && cd .. \
-    && rm -rf patchelf-0.10*
-
-RUN yum install -y python3 python3-devel \
-    && pip3 install requests
-
-RUN wget https://github.com/protocolbuffers/protobuf/releases/download/v3.11.2/protobuf-all-3.11.2.tar.gz && \
-    tar zxf protobuf-all-3.11.2.tar.gz && \
-    cd protobuf-3.11.2 && \
-    ./configure && make -j4 && make install && \
-    make clean && \
-    cd .. && rm -rf protobuf-*
-
-RUN yum -y update >/dev/null \
-    && yum -y install dnf >/dev/null \
-    && yum -y install dnf-plugins-core >/dev/null \
-    && dnf copr enable alonid/llvm-3.8.0 -y \
-    && dnf install llvm-3.8.0 clang-3.8.0 compiler-rt-3.8.0 -y \
-    && echo 'export PATH=/opt/llvm-3.8.0/bin:$PATH' >> /root/.bashrc
-
-RUN yum install -y java \
-    && wget http://repos.fedorapeople.org/repos/dchen/apache-maven/epel-apache-maven.repo -O /etc/yum.repos.d/epel-apache-maven.repo \
-    && yum install -y apache-maven
-
-RUN yum install -y lsof