diff --git a/docker/cuda/7.5/Dockerfile b/docker/cuda/7.5/Dockerfile new file mode 100644 index 000000000000..ff0b0bbb2cd6 --- /dev/null +++ b/docker/cuda/7.5/Dockerfile @@ -0,0 +1,25 @@ +FROM nvidia/cuda:7.5-cudnn5-devel +MAINTAINER Qingsong Liu + +RUN apt-get update && apt-get install -y \ + git \ + libopenblas-dev \ + libopencv-dev \ + python-dev \ + python-numpy \ + python-setuptools \ + wget \ + python-pip \ + unzip + +RUN cd /root && git clone --recursive https://github.com/dmlc/mxnet && cd mxnet && \ + cp make/config.mk . && \ + sed -i 's/USE_BLAS = atlas/USE_BLAS = openblas/g' config.mk && \ + sed -i 's/USE_CUDA = 0/USE_CUDA = 1/g' config.mk && \ + sed -i 's/USE_CUDA_PATH = NONE/USE_CUDA_PATH = \/usr\/local\/cuda/g' config.mk && \ + sed -i 's/USE_CUDNN = 0/USE_CUDNN = 1/g' config.mk && \ + make -j"$(nproc)" + +ENV PYTHONPATH /root/mxnet/python + +WORKDIR /root/mxnet diff --git a/docker/cuda/8.0/Dockerfile b/docker/cuda/8.0/Dockerfile new file mode 100644 index 000000000000..c67375576f03 --- /dev/null +++ b/docker/cuda/8.0/Dockerfile @@ -0,0 +1,25 @@ +FROM nvidia/cuda:8.0-cudnn5-devel +MAINTAINER Qingsong Liu + +RUN apt-get update && apt-get install -y \ + git \ + libopenblas-dev \ + libopencv-dev \ + python-dev \ + python-numpy \ + python-setuptools \ + wget \ + python-pip \ + unzip + +RUN cd /root && git clone --recursive https://github.com/dmlc/mxnet && cd mxnet && \ + cp make/config.mk . && \ + sed -i 's/USE_BLAS = atlas/USE_BLAS = openblas/g' config.mk && \ + sed -i 's/USE_CUDA = 0/USE_CUDA = 1/g' config.mk && \ + sed -i 's/USE_CUDA_PATH = NONE/USE_CUDA_PATH = \/usr\/local\/cuda/g' config.mk && \ + sed -i 's/USE_CUDNN = 0/USE_CUDNN = 1/g' config.mk && \ + make -j"$(nproc)" + +ENV PYTHONPATH /root/mxnet/python + +WORKDIR /root/mxnet diff --git a/docker/cuda/Dockerfile b/docker/cuda/Dockerfile deleted file mode 100644 index fff84352bf16..000000000000 --- a/docker/cuda/Dockerfile +++ /dev/null @@ -1,17 +0,0 @@ -FROM nvidia/cuda:7.5-cudnn4-devel -MAINTAINER Mu Li - -# install the core library -RUN apt-get update && apt-get install -y build-essential git libopenblas-dev libopencv-dev -RUN git clone --recursive https://github.com/dmlc/mxnet/ && cd mxnet && \ - cp make/config.mk . && \ - echo "USE_CUDA=1" >>config.mk && \ - echo "USE_CUDA_PATH=/usr/local/cuda" >>config.mk && \ - echo "USE_CUDNN=1" >>config.mk && \ - echo "USE_BLAS=openblas" >>config.mk && \ - make -j$(nproc) ADD_LDFLAGS=-L/usr/local/cuda/lib64/stubs -ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:$LD_LIBRARY_PATH - -# python pakcage -RUN apt-get install -y python-numpy wget unzip -ENV PYTHONPATH /mxnet/python