dmlc--dgl
cdf7334cc6
* Test CPP branch CI (#2) * Fix batching node-only graphs (#62) * fixing batching with graphs with no edges * oops forgot test * fix readme * Docker and Jenkins (#1) * docker ci cpu * install python packages * docker ci gpu * add readme * use dgl cpu image * run command in container as root * use python3 * fix test case * remove nose from docker file * docker folder readme * parallelize cpu and gpu * top level stages * comment out python2 related installation * fix * remove igraph * building for cpp * change building order * export env in test stage * withEnv * run docker container as root * fix test cases * fix test cases * minor * remove old build
23 行
864 B
Docker
23 行
864 B
Docker
# CI docker GPU env
|
|
FROM nvidia/cuda:9.0-cudnn7-devel
|
|
|
|
# Base scripts
|
|
RUN apt-get update --fix-missing
|
|
|
|
COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh
|
|
RUN bash /install/ubuntu_install_core.sh
|
|
|
|
COPY install/ubuntu_install_python.sh /install/ubuntu_install_python.sh
|
|
RUN bash /install/ubuntu_install_python.sh
|
|
|
|
COPY install/ubuntu_install_python_package.sh /install/ubuntu_install_python_package.sh
|
|
RUN bash /install/ubuntu_install_python_package.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}
|