TensorFlow : Docker イメージ (CPU) を利用する2020/07/24 |
機械学習ライブラリー, TensorFlow をインストールします。
当例では、TensorFlow 公式の Docker イメージをダウンロードして、コンテナーから TensorFlow を利用します。
Docker イメージは GPU サポート無しの CPU オンリーのイメージを使用します。 |
|
[1] | |
[2] | TensorFlow Docker (CPU) の利用例です。 |
# TensorFlow 2.0 with Python3 イメージを Pull [cent@dlp ~]$ docker pull tensorflow/tensorflow:2.0.0-py3
docker images REPOSITORY TAG IMAGE ID CREATED SIZE tensorflow/tensorflow 2.0.0-py3 90f5cb97b18f 9 months ago 1.07GB # コンテナー実行 [cent@dlp ~]$ docker run --rm tensorflow/tensorflow:2.0.0-py3 \ python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))" 2020-07-23 22:08:51.520138: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 2801845000 Hz 2020-07-23 22:08:51.520702: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x37ec9f0 executing computations on platform Host. Devices: 2020-07-23 22:08:51.523989: I tensorflow/compiler/xla/service/service.cc:175] StreamExecutor device (0): Host, Default Version tf.Tensor(1521.4528, shape=(), dtype=float32) import tensorflow as tf hello = tf.constant('Hello, TensorFlow World!') tf.print(hello) docker run --rm -v $PWD:/tmp -w /tmp tensorflow/tensorflow:2.0.0-py3 python ./hello_tensorflow.py 2020-07-23 22:10:02.484191: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 2801845000 Hz 2020-07-23 22:10:02.486266: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x38f12a0 executing computations on platform Host. Devices: 2020-07-23 22:10:02.486299: I tensorflow/compiler/xla/service/service.cc:175] StreamExecutor device (0): Host, Default Version Hello, TensorFlow World! |
[3] | Jupyter Notebook を含むイメージの利用例です。 |
# TensorFlow 2.0 with Python3/Jupyter イメージを Pull [cent@dlp ~]$ docker pull tensorflow/tensorflow:2.0.0-py3-jupyter
docker images REPOSITORY TAG IMAGE ID CREATED SIZE tensorflow/tensorflow 2.0.0-py3-jupyter c652a4fc8a4f 9 months ago 1.22GB tensorflow/tensorflow 2.0.0-py3 90f5cb97b18f 9 months ago 1.07GB # コンテナーをデーモンとして実行 [cent@dlp ~]$ docker run -dt -p 8888:8888 tensorflow/tensorflow:2.0.0-py3-jupyter 4b65c77143ea83773db64031dc73eddb8b31c71c74e3506b700aaa3a523af9b5[cent@dlp ~]$ docker ps CONTAINER ID IMAGE COMMAND CREATED STATUS PORTS NAMES 4b65c77143ea tensorflow/tensorflow:2.0.0-py3-jupyter "bash -c 'source /et…" 38 seconds ago Up 37 seconds 0.0.0.0:8888->8888/tcp suspicious_easley # URL 確認 [cent@dlp ~]$ docker exec 4b65c77143ea bash -c "jupyter notebook list" Currently running servers: http://0.0.0.0:8888/?token=7357a472cdce008e59298c69b20731f1f6ed0bc3f8529a8f :: /tf |
表示された URL にアクセスすると Jupyter Notebook が利用できます。 |
Sponsored Link |
|