Commit c362ded3 authored by Daniele Venzano's avatar Daniele Venzano

Add GPU image

parent 9b7da74b
Pipeline #10367 failed with stages
in 8 minutes and 10 seconds
ARG VERSION=test
FROM zapps/python:${VERSION}
RUN apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
RUN wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.1.85-1_amd64.deb && \
dpkg -i cuda-repo-ubuntu1604_9.1.85-1_amd64.deb && \
rm cuda-repo-ubuntu1604_9.1.85-1_amd64.deb
RUN wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb && \
dpkg -i nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb && \
rm nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
RUN apt-get update && \
apt-get -y install cuda9.0 cuda-cublas-9-0 cuda-cufft-9-0 cuda-curand-9-0 \
cuda-cusolver-9-0 cuda-cusparse-9-0 libcudnn7=7.1.4.18-1+cuda9.0 \
libnccl2=2.2.13-1+cuda9.0 cuda-command-line-tools-9-0 && \
apt-get clean
RUN pip install --upgrade pip \
notebook \
jupyterlab \
tensorflow-gpu==1.10.1 \
tensorboard==1.10.0 \
pandas \
matplotlib \
scipy \
seaborn \
scikit-learn \
scikit-image \
sympy \
cython \
patsy \
statsmodels \
cloudpickle \
dill \
numba \
bokeh \
torch==0.4.1.post2 \
torchvision
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment