-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
🔙 bring back cuda 10.0 for tensorflow 1.x (#52)
- Loading branch information
Showing
3 changed files
with
125 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,89 @@ | ||
| # Distributed under the terms of the Modified BSD License. | ||
|
|
||
| # CUDA base: https://gitlab.com/nvidia/container-images/cuda/-/tree/master/dist/ubuntu18.04/10.0/base/Dockerfile | ||
| # CUDA runtime: https://gitlab.com/nvidia/container-images/cuda/-/tree/master/dist/ubuntu18.04/10.0/runtime/Dockerfile | ||
| # CUDNN runtime: https://gitlab.com/nvidia/container-images/cuda/-/tree/master/dist/ubuntu18.04/10.0/runtime/cudnn7/Dockerfile | ||
| # CUDA devel: https://gitlab.com/nvidia/container-images/cuda/-/tree/master/dist/ubuntu18.04/10.0/devel/Dockerfile | ||
| # CUDNN devel https://gitlab.com/nvidia/container-images/cuda/-/tree/master/dist/ubuntu18.04/10.0/devel/cudnn7/Dockerfile | ||
|
|
||
| ARG repository | ||
| FROM ${repository}:base | ||
|
|
||
| LABEL maintainer="haobibo@gmail.com" | ||
|
|
||
| ARG ARG_CUDA_RUNTIME=true | ||
| ARG ARG_CUDNN_RUNTIME=true | ||
| ARG ARG_CUDA_DEVEL=true | ||
| ARG ARG_CUDNN_DEVEL=true | ||
|
|
||
| ENV CUDA_VER 10.0 | ||
| ENV CUDA_VERSION ${CUDA_VER}.130 | ||
| ENV CUDA_PKG_VERSION 10-0=$CUDA_VERSION-1 | ||
| ENV NCCL_VERSION 2.4.8 | ||
| ENV CUDNN_VERSION 7.6.5.32 | ||
| ENV NVIDIA_REQUIRE_CUDA "cuda>=${CUDA_VER}" | ||
|
|
||
| ENV NVIDIA_VISIBLE_DEVICES=all \ | ||
| NVIDIA_DRIVER_CAPABILITIES=compute,utility \ | ||
| NVIDIA_REQUIRE_CUDA="cuda>=10.0 brand=tesla,driver>=384,driver<385 brand=tesla,driver>=410,driver<411" \ | ||
| PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} \ | ||
| LIBRARY_PATH=/usr/local/cuda/lib64/stubs \ | ||
| LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 | ||
|
|
||
| LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}" | ||
|
|
||
| # Installing CUDA base | ||
| RUN wget -qO- "https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub" | apt-key add - \ | ||
| && echo "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list \ | ||
| && echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list \ | ||
| && apt-get update \ | ||
| && apt-get install -y --no-install-recommends \ | ||
| cuda-cudart-$CUDA_PKG_VERSION cuda-compat-10-0 \ | ||
| && ln -s cuda-$CUDA_VER /usr/local/cuda \ | ||
| && echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf \ | ||
| && echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf | ||
|
|
||
| # If installing CUDA runtime | ||
| RUN ${ARG_CUDA_RUNTIME:-false} \ | ||
| && apt-get install -y --no-install-recommends \ | ||
| cuda-libraries-$CUDA_PKG_VERSION cuda-nvtx-$CUDA_PKG_VERSION libnccl2=$NCCL_VERSION-1+cuda$CUDA_VER \ | ||
| && apt-mark hold libnccl2 \ | ||
| || true | ||
|
|
||
| # If installing CUDNN runtime | ||
| RUN ${ARG_CUDNN_RUNTIME:false} \ | ||
| && apt-get install -y --no-install-recommends \ | ||
| libcudnn7=$CUDNN_VERSION-1+cuda$CUDA_VER \ | ||
| && apt-mark hold libcudnn7 \ | ||
| || true | ||
|
|
||
| # If installing CUDA devel | ||
| RUN ${ARG_CUDA_DEVEL:false} \ | ||
| && apt-get install -y --no-install-recommends \ | ||
| cuda-libraries-dev-$CUDA_PKG_VERSION cuda-nvml-dev-$CUDA_PKG_VERSION \ | ||
| cuda-minimal-build-$CUDA_PKG_VERSION cuda-command-line-tools-$CUDA_PKG_VERSION \ | ||
| libnccl-dev=$NCCL_VERSION-1+cuda$CUDA_VER \ | ||
| || true | ||
|
|
||
| # If installing CUDNN devel | ||
| RUN ${ARG_CUDNN_DEVEL:false} \ | ||
| && apt-get install -y --no-install-recommends \ | ||
| libcudnn7=$CUDNN_VERSION-1+cuda$CUDA_VER libcudnn7-dev=$CUDNN_VERSION-1+cuda$CUDA_VER \ | ||
| || true | ||
|
|
||
| # Install Utilities `nvtop` | ||
| RUN cd /tmp \ | ||
| && apt-get -y update --fix-missing && apt-get -qq install -y --no-install-recommends libncurses5-dev \ | ||
| && git clone https://github.com/Syllo/nvtop.git \ | ||
| && mkdir -p nvtop/build && cd nvtop/build \ | ||
| && LIB_PATH=`find / -name "libnvidia-ml*" 2>/dev/null` \ | ||
| && cmake .. -DCMAKE_LIBRARY_PATH="`dirname $LIB_PATH`" .. \ | ||
| && make && make install \ | ||
| && apt-get -qq remove -y libncurses5-dev | ||
|
|
||
| # Clean up and display components version information... | ||
| RUN source /opt/utils/script-utils.sh \ | ||
| && conda install -yq python=3.7 \ | ||
| && install__clean && cd \ | ||
| && echo "@ Version of image: building finished at:" `date` `uname -a` \ | ||
| && echo "@ System environment variables:" `printenv` |