配置horovod环境dockerfile

最新推荐文章于 2023-03-16 22:09:36 发布

shicript

最新推荐文章于 2023-03-16 22:09:36 发布

阅读量369

点赞数

分类专栏： tensorflow 深度学习 horovod

本文链接：https://blog.csdn.net/qq_33599158/article/details/107582355

版权

这篇博客详细介绍了如何通过Dockerfile来配置Horovod的运行环境，为深度学习的分布式训练提供基础。

摘要由CSDN通过智能技术生成

dockerfile文件如下：

FROM nvidia/cuda:10.0-devel-ubuntu18.04

# TensorFlow version is tightly coupled to CUDA and cuDNN so it should be selected carefully
ENV TENSORFLOW_VERSION=1.14.0
ENV PYTORCH_VERSION=1.2.0
ENV TORCHVISION_VERSION=0.4.0
ENV CUDNN_VERSION=7.6.0.64-1+cuda10.0
ENV NCCL_VERSION=2.4.7-1+cuda10.0
ENV MXNET_VERSION=1.5.0
ENV LC_ALL=C.UTF-8

# Python 2.7 or 3.6 is supported by Ubuntu Bionic out of the box
ARG python=3.6
ENV PYTHON_VERSION=${python}

# Set default shell to /bin/bash
SHELL ["/bin/bash", "-cu"]

RUN apt-get update && apt-get install -y --allow-downgrades --allow-change-held-packages --no-install-recommends \
        build-essential \
        cmake \
        g++-4.8 \
        git \
        curl \
        vim \
        wget \
        ca-certificates \
        libcudnn7=${CUDNN_VERSION} \
        libnccl2=${NCCL_VERSION} \
        l