Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
25 lines (19 sloc) 831 Bytes
FROM ubuntu:14.04
MAINTAINER Traun Leyden <traun.leyden@gmail.com>
# A docker container with the Nvidia kernel module and CUDA drivers installed
ENV CUDA_RUN http://developer.download.nvidia.com/compute/cuda/6_5/rel/installers/cuda_6.5.14_linux_64.run
RUN apt-get update && apt-get install -q -y \
wget \
build-essential \
module-init-tools
RUN cd /opt && \
wget $CUDA_RUN && \
chmod +x *.run && \
mkdir nvidia_installers && \
./cuda_6.5.14_linux_64.run -extract=`pwd`/nvidia_installers && \
cd nvidia_installers && \
./NVIDIA-Linux-x86_64-340.29.run -s -N --no-kernel-module && \
./cuda-linux64-rel-6.5.14-18749181.run -noprompt
# Ensure the CUDA libs and binaries are in the correct environment variables
ENV LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-6.5/lib64
ENV PATH=$PATH:/usr/local/cuda-6.5/bin