Dockerfile.ci_cpu 1.34 KB
Newer Older
Lingfan Yu's avatar
Lingfan Yu committed
1
2
# CI docker CPU env
# Adapted from github.com/dmlc/tvm/docker/Dockerfile.ci_cpu
3
4
5
6
FROM ubuntu:18.04

ENV TZ=US
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
Lingfan Yu's avatar
Lingfan Yu committed
7
8
9
10
11
12

RUN apt-get update --fix-missing

COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh
RUN bash /install/ubuntu_install_core.sh

13
14
15
COPY install/ubuntu_install_build.sh /install/ubuntu_install_build.sh
RUN bash /install/ubuntu_install_build.sh

VoVAllen's avatar
VoVAllen committed
16
17
18
# python
COPY install/ubuntu_install_conda.sh /install/ubuntu_install_conda.sh
RUN bash /install/ubuntu_install_conda.sh
Minjie Wang's avatar
Minjie Wang committed
19

VoVAllen's avatar
VoVAllen committed
20
ENV CONDA_ALWAYS_YES="true"
Minjie Wang's avatar
Minjie Wang committed
21

VoVAllen's avatar
VoVAllen committed
22
COPY install/conda_env/torch_cpu.yml /install/conda_env/torch_cpu.yml
23
COPY install/conda_env/torch_cpu_pip.txt /install/conda_env/torch_cpu_pip.txt
VoVAllen's avatar
VoVAllen committed
24
RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/torch_cpu.yml"]
Lingfan Yu's avatar
Lingfan Yu committed
25

VoVAllen's avatar
VoVAllen committed
26
27
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"]
28

VoVAllen's avatar
VoVAllen committed
29
30
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"]
Minjie Wang's avatar
Minjie Wang committed
31

32
33
34
35
36
37
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