This repository has been archived, as we dont maintain it's content anymore, there is no new home for the code, feel free to contained any of the contributors!
Tensorflow in Containers
About
This respository has Dockerfiles which have tensorflow wheel files built by Red Hat AICoE team. They can be used for testing and experimenting with tensorflow on following OS :
Fedora27
Fedora28
Fedora29
Fedora30
CentOS7
RHEL7.5
RHEL8
For application development use S2I Images & Tensorflow wheel files from https://github.com/AICoE/tensorflow-wheels/releases.
NOTE: for RHEL7.5
& RHEL8
you need a system with RHEL Subscription enabled.
Usage
Build Image
docker build --build-arg "TF_URL=${TF_URL}" -t aicoe/tf-in-container:1.13.1-fedora30 -f Dockerfile .
TF_URL value should be taken from https://github.com/AICoE/tensorflow-wheels/releases.
Use Image
docker run -it quay.io/aicoe/tf-in-container:1.13.1-fedora30 bash
Example
$ docker run -it quay.io/aicoe/tf-in-container:1.13.1-fedora30 bash
[root@94fc148a10de /]# python3 -c "$TEST_CMD"
Device mapping: no known devices.
2019-06-24 15:21:10.309103: I tensorflow/core/common_runtime/direct_session.cc:317] Device mapping:
MatMul: (MatMul): /job:localhost/replica:0/task:0/device:CPU:0
2019-06-24 15:21:10.311608: I tensorflow/core/common_runtime/placer.cc:1059] MatMul: (MatMul)/job:localhost/replica:0/task:0/device:CPU:0
a: (Const): /job:localhost/replica:0/task:0/device:CPU:0
2019-06-24 15:21:10.311692: I tensorflow/core/common_runtime/placer.cc:1059] a: (Const)/job:localhost/replica:0/task:0/device:CPU:0
b: (Const): /job:localhost/replica:0/task:0/device:CPU:0
2019-06-24 15:21:10.311816: I tensorflow/core/common_runtime/placer.cc:1059] b: (Const)/job:localhost/replica:0/task:0/device:CPU:0
[[22\. 28.]
[49\. 64.]]
[root@94fc148a10de /]#