mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-03 19:34:33 +08:00
42 lines
1.5 KiB
Docker
42 lines
1.5 KiB
Docker
# CI docker CPU env
|
|
# Adapted from github.com/dmlc/tvm/docker/Dockerfile.ci_cpu
|
|
FROM ubuntu:20.04
|
|
|
|
ENV TZ=US
|
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
|
|
|
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_build.sh /install/ubuntu_install_build.sh
|
|
RUN bash /install/ubuntu_install_build.sh
|
|
|
|
# tcmalloc
|
|
RUN apt-get install -y libgoogle-perftools4
|
|
ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4:$LD_PRELOAD
|
|
|
|
# python
|
|
COPY install/ubuntu_install_conda.sh /install/ubuntu_install_conda.sh
|
|
RUN bash /install/ubuntu_install_conda.sh
|
|
|
|
ENV CONDA_ALWAYS_YES="true"
|
|
|
|
COPY install/conda_env/torch_cpu.yml /install/conda_env/torch_cpu.yml
|
|
COPY install/conda_env/torch_cpu_pip.txt /install/conda_env/torch_cpu_pip.txt
|
|
RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/torch_cpu.yml"]
|
|
|
|
COPY install/conda_env/tensorflow_cpu.yml /install/conda_env/tensorflow_cpu.yml
|
|
RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/tensorflow_cpu.yml"]
|
|
|
|
COPY install/conda_env/mxnet_cpu.yml /install/conda_env/mxnet_cpu.yml
|
|
RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/mxnet_cpu.yml"]
|
|
|
|
ENV CONDA_ALWAYS_YES=
|
|
|
|
# SSH
|
|
RUN ["/bin/bash", "-i", "-c", "ssh-keygen -f ~/.ssh/id_rsa -N ''"]
|
|
RUN ["/bin/bash", "-i", "-c", "cat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys"]
|
|
ENTRYPOINT service ssh restart && bash
|