2017-08-28 04:26:33 -04:00
|
|
|
FROM nvidia/cuda:8.0-runtime-ubuntu16.04
|
|
|
|
|
|
|
|
|
|
LABEL com.nvidia.volumes.needed="nvidia_driver"
|
|
|
|
|
|
|
|
|
|
ENV PATH /usr/local/nvidia/bin:${PATH}
|
|
|
|
|
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}
|
|
|
|
|
|
2017-12-05 07:08:04 -05:00
|
|
|
RUN apt-get update&&apt-get install -y git software-properties-common python-software-properties python-setuptools python3-setuptools
|
2017-09-04 14:31:34 -04:00
|
|
|
|
2017-08-28 04:26:33 -04:00
|
|
|
RUN git clone --recursive http://git.tiker.net/trees/pycuda.git
|
|
|
|
|
RUN git clone --recursive https://github.com/twobombs/cuburn.git
|
|
|
|
|
|
2017-12-05 06:34:26 -05:00
|
|
|
RUN cd /pycuda&&./configure.py&&make&&make install
|
2017-08-28 04:26:33 -04:00
|
|
|
|
2017-12-05 05:03:41 -05:00
|
|
|
RUN add-apt-repository universe && apt-get update && export DEBIAN_FRONTEND=noninteractive && apt-get install -y libboost-all-dev python-pycuda python-pip && apt-get clean all
|
2017-08-28 04:26:33 -04:00
|
|
|
|
|
|
|
|
RUN pip install numpy scipy
|
