experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix for 'Horovod>0.19.5' installation

Browse Source
This commit is contained in:
Abolfazl Shahbazi 2020-12-17 10:59:42 -08:00
parent 7dad9545e5
commit 0b7502288d
11 changed files with 15 additions and 2 deletions

2
tensorflow/tools/dockerfiles/dockerfiles/onednn/centos-8-mpi-horovod-jupyter.Dockerfile
View File

@ -88,8 +88,10 @@ ARG HOROVOD_WITH_TENSORFLOW=1
ARG HOROVOD_VERSION=
RUN yum update -y && yum install -y \
cmake \
gcc \
gcc-c++ \
make \
python36-devel && \
yum clean all

2
tensorflow/tools/dockerfiles/dockerfiles/onednn/centos-8-mpi-horovod.Dockerfile
View File

@ -88,8 +88,10 @@ ARG HOROVOD_WITH_TENSORFLOW=1
ARG HOROVOD_VERSION=
RUN yum update -y && yum install -y \
cmake \
gcc \
gcc-c++ \
make \
python36-devel && \
yum clean all

2
tensorflow/tools/dockerfiles/dockerfiles/onednn/centos-8-mpich-horovod-jupyter.Dockerfile
View File

@ -87,8 +87,10 @@ ARG HOROVOD_WITH_TENSORFLOW=1
ARG HOROVOD_VERSION=
RUN yum update -y && yum install -y \
cmake \
gcc \
gcc-c++ \
make \
python36-devel && \
yum clean all

2
tensorflow/tools/dockerfiles/dockerfiles/onednn/centos-8-mpich-horovod.Dockerfile
View File

@ -87,8 +87,10 @@ ARG HOROVOD_WITH_TENSORFLOW=1
ARG HOROVOD_VERSION=
RUN yum update -y && yum install -y \
cmake \
gcc \
gcc-c++ \
make \
python36-devel && \
yum clean all

1
tensorflow/tools/dockerfiles/dockerfiles/onednn/ubuntu-18.04-mpi-horovod-jupyter.Dockerfile
View File

@ -85,6 +85,7 @@ ARG HOROVOD_VERSION=
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
build-essential \
cmake \
g++-8 \
gcc-8 \
python3-dev

1
tensorflow/tools/dockerfiles/dockerfiles/onednn/ubuntu-18.04-mpi-horovod.Dockerfile
View File

@ -85,6 +85,7 @@ ARG HOROVOD_VERSION=
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
build-essential \
cmake \
g++-8 \
gcc-8 \
python3-dev

1
tensorflow/tools/dockerfiles/dockerfiles/onednn/ubuntu-18.04-mpich-horovod-jupyter.Dockerfile
View File

@ -81,6 +81,7 @@ ARG HOROVOD_VERSION=
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
build-essential \
cmake \
g++-8 \
gcc-8 \
python3-dev

1
tensorflow/tools/dockerfiles/dockerfiles/onednn/ubuntu-18.04-mpich-horovod.Dockerfile
View File

@ -81,6 +81,7 @@ ARG HOROVOD_VERSION=
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
build-essential \
cmake \
g++-8 \
gcc-8 \
python3-dev

2
tensorflow/tools/dockerfiles/partials/onednn/centos/horovod.partial.Dockerfile
View File

@ -5,8 +5,10 @@ ARG HOROVOD_WITH_TENSORFLOW=1
ARG HOROVOD_VERSION=
RUN yum update -y && yum install -y \
cmake \
gcc \
gcc-c++ \
make \
python36-devel && \
yum clean all

1
tensorflow/tools/dockerfiles/partials/onednn/ubuntu/horovod.partial.Dockerfile
View File

@ -6,6 +6,7 @@ ARG HOROVOD_VERSION=
RUN apt-get update && apt-get install -y --no-install-recommends --fix-missing \
build-essential \
cmake \
g++-8 \
gcc-8 \
python3-dev

2
tensorflow/tools/dockerfiles/spec.yml
View File

@ -351,7 +351,6 @@ slice_sets:
args:
- CENTOS_VERSION=8
- PYTHON=python3
- HOROVOD_VERSION=0.19.5
- TF_PACKAGE=intel-tensorflow
devel-onednn-mpi-horovod:
@ -499,7 +498,6 @@ slice_sets:
args:
- CENTOS_VERSION=8
- PYTHON=python3
- HOROVOD_VERSION=0.19.5
- TF_PACKAGE=intel-tensorflow
devel-onednn-mpich-horovod: