Deep Learning Reference Stack with Tensorflow and Intel® oneAPI Deep Neural Network Library (oneDNN)
Building Locally
We have created a set of Dockerfiles that allow you to build DLRS Tensorflow with different configurations. You will be able to pick the OS (Ubuntu or Centos), then pick the flavour, either core
, which is a leaner image with just the framework, and full
which includes the framework and extra features. Please refer to the table below to see all features and configurations.
Framework | Features | |
---|---|---|
Core | X | |
Full | X | X |
Framework:
TensorFlow* 1.15.3 or
TensorFlow* 2.4.0
Features:
Transformers
Horovod
Seldon core 1.2.0
NOTE: We recommend you enable Docker Buildkit to have concurrent dependency resolution and automatic garbage collection. Docker Buildkit has been integrated in Docker since 18.06, if you have an older version, please ignore this note.
Enable Docker Buildkit (see note above)
export DOCKER_BUILDKIT=1
Building Ubuntu based DLRS
Tensorflow 1.15.*
Core:
docker build --build_arg tf_ver=tf1 -t dlrs-tensorflow-ubuntu:v0.9.0 -f Dockerfile.ubuntu --target tf_core .
Full:
docker build --build_arg tf_ver=tf1 -t dlrs-tensorflow-ubuntu:v0.9.0 -f Dockerfile.ubuntu --target tf_full.
Tensorflow 2.4.0
Core:
docker build --build_arg tf_ver=tf2 -t dlrs-tensorflow2-ubuntu:v0.9.0 -f Dockerfile.ubuntu --target tf_core .
Full:
docker build --build_arg tf_ver=tf2 -t dlrs-tensorflow2-ubuntu:v0.9.0 -f Dockerfile.ubuntu --target tf_full.
Building Centos based DLRS
Tensorflow 1.15.0
Core:
docker build --build_arg tf_ver=tf1 -t dlrs-tensorflow-centos:v0.9.0 -f Dockerfile.centos --target tf_core .
Full:
docker build --build_arg tf_ver=tf1 -t dlrs-tensorflow-centos:v0.9.0 -f Dockerfile.centos --target tf_full.
Tensorflow 2.4.0
Core:
docker build --build_arg tf_ver=tf2 -t dlrs-tensorflow2-centos:v0.9.0 -f Dockerfile.centos --target tf_core .
Full:
docker build --build_arg tf_ver=tf2 -t dlrs-tensorflow2-centos:v0.9.0 -f Dockerfile.centos --target tf_full.