OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Branch: release-1.9
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'release-1.9'
${ noResults }
MegEngine/ci/docker_env/Dockerfile

51 lines
2.7 kB
Raw Permalink Blame History 

  1. FROM nvidia/cuda:10.1-devel-ubuntu18.04

  2. RUN rm /etc/apt/sources.list.d/cuda.list

  3. 

  4. RUN apt-get update && apt-get install -y --no-install-recommends \

  5.     autoconf \

  6.     autotools-dev \

  7.     automake \

  8.     clang-6.0 \

  9.     curl \

  10.     git-lfs \

  11.     libtool \

  12.     libpcre3-dev \

  13.     llvm-6.0-dev \

  14.     openssh-client \

  15.     openssh-server \

  16.     pkg-config \

  17.     python-pip \

  18.     python3-pip \

  19.     python3-dev \

  20.     python-numpy \

  21.     python3-numpy \

  22.     python3-setuptools \

  23.     software-properties-common \

  24.     swig \

  25.     vim \

  26.     wget \

  27.     libgl1-mesa-glx \

  28.     libsm6 \

  29.     libxext6 \

  30.     zlib1g-dev \

  31.     # GitLab Runner need Git 2.18 or higher to create a local Git repository

  32.     && add-apt-repository ppa:git-core/ppa -y && apt-get install --no-install-recommends -y git \

  33.     && rm -rf /var/lib/apt/lists/*

  34. 

  35. RUN cd /tmp ; wget https://cmake.org/files/v3.15/cmake-3.15.2.tar.gz;tar -xzvf cmake-3.15.2.tar.gz;cd cmake-3.15.2;./configure; make -j32; make install

  36. 

  37. RUN git lfs install

  38. 

  39. RUN pip3 install --upgrade pip

  40. 

  41. # TODO: set following envs in github environment.

  42. ENV CUDA_ROOT_DIR=/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1 \

  43.   TRT_ROOT_DIR=/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/TensorRT-6.0.1.5 \

  44.   TENSORRT_ROOT_DIR=${TRT_ROOT_DIR} \

  45.   CUDNN_ROOT_DIR=/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cudnn-v7.6.3 \

  46.   PATH=/usr/bin:${CUDA_ROOT_DIR}/bin:${CUDA_ROOT_DIR}/nsight-compute-2019.4.0:$PATH \

  47.   LIBRARY_PATH=${LIBRARY_PATH}:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1/lib:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1/lib64:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1/lib/stubs:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1/lib64/stubs:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cudnn-v7.6.3/lib:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cudnn-v7.6.3/lib64:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/TensorRT-6.0.1.5/lib:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/TensorRT-6.0.1.5/lib64 \

  48.   LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cudnn-v7.6.3/lib:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cudnn-v7.6.3/lib64:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/TensorRT-6.0.1.5/lib:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/TensorRT-6.0.1.5/lib64  \

  49.   CPATH=${CPATH}:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1/include:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cudnn-v7.6.3/include:/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/TensorRT-6.0.1.5/include \

  50.   CUDA_BIN_PATH=/usr/local/cuda-10.1-cudnn-7.6.3-trt-6.0.1.5-libs/cuda-10.1