|
|
@@ -0,0 +1,39 @@ |
|
|
|
FROM nvidia/cuda:10.1-devel-ubuntu18.04 |
|
|
|
RUN rm /etc/apt/sources.list.d/cuda.list |
|
|
|
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \ |
|
|
|
autoconf \ |
|
|
|
autotools-dev \ |
|
|
|
automake \ |
|
|
|
clang-6.0 \ |
|
|
|
curl \ |
|
|
|
git-lfs \ |
|
|
|
libtool \ |
|
|
|
libpcre3-dev \ |
|
|
|
llvm-6.0-dev \ |
|
|
|
openssh-client \ |
|
|
|
openssh-server \ |
|
|
|
pkg-config \ |
|
|
|
python-pip \ |
|
|
|
python3-pip \ |
|
|
|
python3-dev \ |
|
|
|
python-numpy \ |
|
|
|
python3-numpy \ |
|
|
|
python3-setuptools \ |
|
|
|
software-properties-common \ |
|
|
|
swig \ |
|
|
|
vim \ |
|
|
|
wget \ |
|
|
|
zlib1g-dev \ |
|
|
|
# GitLab Runner need Git 2.18 or higher to create a local Git repository |
|
|
|
&& add-apt-repository ppa:git-core/ppa -y && apt-get install --no-install-recommends -y git \ |
|
|
|
&& rm -rf /var/lib/apt/lists/* |
|
|
|
|
|
|
|
RUN cd /tmp ; wget https://cmake.org/files/v3.14/cmake-3.14.4.tar.gz;tar -xzvf cmake-3.14.4.tar.gz;cd cmake-3.14.4;./configure; make -j32; make install |
|
|
|
|
|
|
|
RUN git lfs install |
|
|
|
|
|
|
|
ENV PATH=${PATH}:/usr/local/cuda/bin \ |
|
|
|
LIBRARY_PATH=${LIBRARY_PATH}:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda/lib/stubs:/usr/local/cuda/lib64/stubs:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib64:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib64 \ |
|
|
|
LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda/lib/stubs:/usr/local/cuda/lib64/stubs:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib64:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib64:/usr/local/cuda/lib64/stubs/libcuda.so:/tmp/build/cuda/dnn/cuda-stub/libcuda.so.1 \ |
|
|
|
CPATH=${CPATH}:/usr/local/cuda/include:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/include:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/include |