modify gpu-dockerfile

This commit is contained in:
wuweikang 2021-02-04 09:37:03 +08:00
parent 6b39c89da7
commit 47ad5bf806
1 changed files with 2 additions and 2 deletions

View File

@ -6,8 +6,8 @@ MAINTAINER leonwanghui <leon.wanghui@huawei.com>
ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5
ENV CMAKE_ROOT_PATH /usr/local/cmake-3.14.1
ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3
ENV PATH ${CMAKE_ROOT_PATH}/bin:/usr/local/bin:$PATH
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}:${PYTHON_ROOT_PATH}/lib
ENV PATH ${OMPI_ROOT_PATH}/bin:${PYTHON_ROOT_PATH}/bin:${CMAKE_ROOT_PATH}/bin:/usr/local/bin:$PATH
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib::${PYTHON_ROOT_PATH}/lib
# Install base tools
RUN apt update \