Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix gpu docker environment paths

This commit is contained in:
yanghaoran 2022-04-27 19:02:55 +08:00
parent 71d55684e1
commit 09da09f216
4 changed files with 12 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/docker/mindspore-gpu/1.7.0/cuda10.1/Dockerfile
View File

@ -5,8 +5,9 @@ MAINTAINER MindSpore Authors
# Set env # Set env
ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5 ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5
ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3 ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3
ENV PATH ${PYTHON_ROOT_PATH}/bin:${OMPI_ROOT_PATH}/bin:/usr/local/bin:$PATH ENV CUDA_PATH /usr/local/cuda
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:$LD_LIBRARY_PATH ENV PATH ${PYTHON_ROOT_PATH}/bin:${OMPI_ROOT_PATH}/bin:${CUDA_PATH}/bin:/usr/local/bin:$PATH
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:${CUDA_PATH}/lib64:$LD_LIBRARY_PATH
# Install base tools # Install base tools
RUN apt update \ RUN apt update \

5
scripts/docker/mindspore-gpu/1.7.0/cuda11.1/Dockerfile
View File

@ -5,8 +5,9 @@ MAINTAINER MindSpore Authors
# Set env # Set env
ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5 ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5
ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3 ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3
ENV PATH ${PYTHON_ROOT_PATH}/bin:${OMPI_ROOT_PATH}/bin:/usr/local/bin:$PATH ENV CUDA_PATH /usr/local/cuda
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:$LD_LIBRARY_PATH ENV PATH ${PYTHON_ROOT_PATH}/bin:${OMPI_ROOT_PATH}/bin:${CUDA_PATH}/bin:/usr/local/bin:$PATH
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:${CUDA_PATH}/lib64:$LD_LIBRARY_PATH
# Install base tools # Install base tools
RUN apt update \ RUN apt update \

5
scripts/docker/mindspore-gpu/devel/Dockerfile
View File

@ -6,8 +6,9 @@ MAINTAINER MindSpore Authors
ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5 ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5
ENV CMAKE_ROOT_PATH /usr/local/cmake-3.14.1 ENV CMAKE_ROOT_PATH /usr/local/cmake-3.14.1
ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3 ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3
ENV PATH ${OMPI_ROOT_PATH}/bin:${PYTHON_ROOT_PATH}/bin:${CMAKE_ROOT_PATH}/bin:/usr/local/bin:$PATH ENV CUDA_PATH /usr/local/cuda
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib::${PYTHON_ROOT_PATH}/lib ENV PATH ${PYTHON_ROOT_PATH}/bin:${OMPI_ROOT_PATH}/bin:${CMAKE_ROOT_PATH}/bin:${CUDA_PATH}/bin:/usr/local/bin:$PATH
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:${PYTHON_ROOT_PATH}:${CUDA_PATH}/lib64:$LD_LIBRARY_PATH
# Install base tools # Install base tools
RUN apt update \ RUN apt update \

5
scripts/docker/mindspore-gpu/runtime/Dockerfile
View File

@ -5,8 +5,9 @@ MAINTAINER MindSpore Authors
# Set env # Set env
ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5 ENV PYTHON_ROOT_PATH /usr/local/python-3.7.5
ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3 ENV OMPI_ROOT_PATH /usr/local/openmpi-4.0.3
ENV PATH ${OMPI_ROOT_PATH}/bin:/usr/local/bin:$PATH ENV CUDA_PATH /usr/local/cuda
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:$LD_LIBRARY_PATH ENV PATH ${PYTHON_ROOT_PATH}/bin:${OMPI_ROOT_PATH}/bin:${CUDA_PATH}/bin:/usr/local/bin:$PATH
ENV LD_LIBRARY_PATH ${OMPI_ROOT_PATH}/lib:${CUDA_PATH}/lib64:$LD_LIBRARY_PATH
# Install base tools # Install base tools
RUN apt update \ RUN apt update \