xgboost/tests/ci_build/Dockerfile.gpu

17 lines
504 B
Docker

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