[GPU-Plugin] Add basic continuous integration for GPU plugin. (#2431)

This commit is contained in:
Rory Mitchell
2017-06-23 02:15:28 +12:00
committed by Yuan (Terry) Tang
parent 2cb51f7097
commit 1899f9e744
6 changed files with 253 additions and 0 deletions

View File

@@ -0,0 +1,16 @@
FROM nvidia/cuda:8.0-devel-ubuntu14.04
RUN apt-get update && apt-get -y upgrade
# CMAKE
RUN sudo apt-get install -y build-essential
RUN apt-get install -y wget
RUN wget http://www.cmake.org/files/v3.5/cmake-3.5.2.tar.gz
RUN tar -xvzf cmake-3.5.2.tar.gz
RUN cd cmake-3.5.2/ && ./configure && make && sudo make install
# BLAS
RUN apt-get install -y libatlas-base-dev
# PYTHON2
RUN apt-get install -y python-setuptools python-pip python-dev unzip gfortran
RUN pip install numpy nose scipy scikit-learn