Dockerfile

FROM ubuntu:15.04
MAINTAINER Majid alDosari



# Setup environment

#for cuda 7 use gcc 4.8>
ENV CUDA_GCC_VER 4.9

ENV CUDA_VER1 7
ENV CUDA_VER2 5
ENV CUDA_VER3 18

#jan. 5, 2016
ENV DRIVER_VER 361.16

#todo: make a check for driver download existence
#installing the driver from cuda does not work
#maybe bc driver in cuda cant install in 4.x kernel
#ENV DRIVER_VER CUDA
#'CUDA' means whatever driver came with CUDA

RUN apt-get -y update && apt-get -y install \
    gcc-${CUDA_GCC_VER} g++-${CUDA_GCC_VER} \
    libssl-dev bc \
    wget curl git make dpkg-dev module-init-tools && \
    mkdir -p /usr/src/kernels && \
    mkdir -p /opt/nvidia && \
    apt-get autoremove && apt-get clean
# Ensure we're using gcc version GCC_VER
RUN update-alternatives --install \
            /usr/bin/gcc gcc /usr/bin/gcc-${CUDA_GCC_VER} 60 \
            --slave \
            /usr/bin/g++ g++ /usr/bin/g++-${CUDA_GCC_VER}
#print gcc ver to check
RUN update-alternatives --config gcc


# Download CUDA
# downloading CUDA JUST to expose nvidia_uvm kernel module
# by running devicequery from samples
# todo: do i have to install the samples?

#different urls for differnt versions. thx nvidia!
#if not caring about cuda
#RUN if [ "${CUDA_VER1}" -ge 7 ] && [ "${DRIVER_VER}" = "CUDA" ]
RUN if [ "${CUDA_VER1}" -ge 7 ] && [ "${CUDA_VER2}" -eq 0 ] ; \
then \
curl http://developer.download.nvidia.com/compute/cuda/\
${CUDA_VER1}_${CUDA_VER2}/\
Prod/local_installers/\
cuda_${CUDA_VER1}.${CUDA_VER2}.${CUDA_VER3}_linux.run \
    > /opt/nvidia/cuda.run ;\
fi
RUN if [ "${CUDA_VER1}" -ge 7 ] && [ "${CUDA_VER2}" -ge 5 ] ; \
then \
curl http://developer.download.nvidia.com/compute/cuda/\
${CUDA_VER1}.${CUDA_VER2}/\
Prod/local_installers/\
cuda_${CUDA_VER1}.${CUDA_VER2}.${CUDA_VER3}_linux.run \
    > /opt/nvidia/cuda.run ;\
fi
#RUN if [ "${CUDA_VER1}" -lt 7 ] && [ "${DRIVER_VER}" = "CUDA" ]
RUN if [ "${CUDA_VER1}" -lt 7 ] ; \
then curl http://developer.download.nvidia.com/compute/cuda/\
${CUDA_VER1}_${CUDA_VER2}/\
rel/installers/\
cuda_${CUDA_VER1}.${CUDA_VER2}.${CUDA_VER3}_linux_64.run \
    > /opt/nvidia/cuda.run ;\
fi


# Download driver
# there is a driver in the cuda download BUT it doesn't work!
# as of CUDA 7.0.28

RUN mkdir -p /opt/nvidia
RUN if [ "$DRIVER_VER" != "CUDA" ]; \
then curl  http://us.download.nvidia.com/XFree86/Linux-x86_64/\
${DRIVER_VER}/\
NVIDIA-Linux-x86_64-${DRIVER_VER}.run \
    > /opt/nvidia/driver.run ; fi

WORKDIR /opt/nvidia
COPY install_nvidia.sh /opt/nvidia/install_nvidia.sh
RUN chmod +x ./install_nvidia.sh
RUN chmod +x ./cuda.run
RUN chmod +x ./driver.run
CMD /opt/nvidia/install_nvidia.sh


# Common stuff
# in case this Dockerfile is used as a base to build images

ONBUILD WORKDIR /opt/nvidia
ONBUILD RUN chmod +x cuda.run
ONBUILD RUN chmod +x driver.run
#so when this dockerfile is called it just does the toolkit ...
#..and (not the kernel module)
#opengl support gives an error (not installed?) but the driver
#install may work anyways
ONBUILD RUN ./driver.run \
	    	--silent \
		--no-kernel-module \
		--no-unified-memory \
		--no-opengl-files
#the samples take space but are a great way to chk cuda
#you could remove --samples
ONBUILD RUN ./cuda.run \
	    	--toolkit \
		--samples \
		--silent

# setup stuff
ONBUILD ENV PATH=/usr/local/cuda/bin:$PATH
ONBUILD ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
ONBUILD RUN echo "/usr/local/cuda/lib64" \
	    	> /etc/ld.so.conf.d/cuda.conf && \
		  ldconfig

# 
ONBUILD WORKDIR /root
ONBUILD CMD /bin/bash