Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "25ed7cb08bbfd703e2dc9a0e88641ce17c74deb0"
Unverified Commit b5afaeac authored by Minjie Wang's avatar Minjie Wang Committed by GitHub
Browse files

[CI] Small change on the MX CPU image (#343)

parent e19cd62e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 10 deletions
docker/Dockerfile.ci_cpu_mxnet
View file @ b5afaeac
# CI docker GPU env
FROM nvidia/cuda:9.0-cudnn7-devel
# CI docker CPU env
FROM ubuntu:16.04
# Base scripts
RUN apt-get update --fix-missing
......@@ -15,11 +15,3 @@ RUN bash /install/ubuntu_install_python_package.sh
COPY install/ubuntu_install_mxnet_cpu.sh /install/ubuntu_install_mxnet_cpu.sh
RUN bash /install/ubuntu_install_mxnet_cpu.sh
# Environment variables
ENV PATH=/usr/local/nvidia/bin:${PATH}
ENV PATH=/usr/local/cuda/bin:${PATH}
ENV CPLUS_INCLUDE_PATH=/usr/local/cuda/include:${CPLUS_INCLUDE_PATH}
ENV C_INCLUDE_PATH=/usr/local/cuda/include:${C_INCLUDE_PATH}
ENV LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LIBRARY_PATH}
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment