diff --git a/tools/Dockerfile.gpu b/tools/Dockerfile.gpu index 427ae83bcb805ec70c1e6d575e84234f17e9fb30..091f4a546b549a3dd53645e78ab49b1cd46bf5b3 100644 --- a/tools/Dockerfile.gpu +++ b/tools/Dockerfile.gpu @@ -10,6 +10,6 @@ RUN yum -y install wget && \ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py && \ python get-pip.py && rm get-pip.py && \ ln -s /usr/local/cuda-9.0/lib64/libcublas.so.9.0 /usr/local/cuda-9.0/lib64/libcublas.so && \ - echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64':$LD_LIBRARY_PATH >> /root/.bashrc && \ + echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> /root/.bashrc && \ ln -s /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7 /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so && \ echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/targets/x86_64-linux/lib:$LD_LIBRARY_PATH' >> /root/.bashrc diff --git a/tools/serving_build.sh b/tools/serving_build.sh index b810e3139803bd363c771c6f655cef6595177dc8..2dc012d640b8ebeaf6d02a841fcba1cbab3302a6 100644 --- a/tools/serving_build.sh +++ b/tools/serving_build.sh @@ -27,7 +27,7 @@ function build_client() { -DPYTHON_EXECUTABLE=$PYTHONROOT/bin/python \ -DCLIENT_ONLY=ON .. check_cmd "make -j2 >/dev/null" - pip install python/dist/paddle_serving_client* >/dev/null + pip install -U python/dist/paddle_serving_client* >/dev/null ;; *) echo "error type" @@ -50,7 +50,7 @@ function build_server() { -DPYTHON_EXECUTABLE=$PYTHONROOT/bin/python \ -DCLIENT_ONLY=OFF .. check_cmd "make -j2 >/dev/null" - pip install python/dist/paddle_serving_server* >/dev/null + pip install -U python/dist/paddle_serving_server* >/dev/null ;; GPU) cmake -DPYTHON_INCLUDE_DIR=$PYTHONROOT/include/python2.7/ \ @@ -59,7 +59,7 @@ function build_server() { -DCLIENT_ONLY=OFF \ -DWITH_GPU=ON .. check_cmd "make -j2 >/dev/null" - pip install python/dist/paddle_serving_server* >/dev/null + pip install -U python/dist/paddle_serving_server* >/dev/null ;; *) echo "error type"