首頁 > 軟體

基於Docker的TensorFlow機器學習框架搭建和範例原始碼解讀

2020-06-16 17:21:05

概述:基於Docker的TensorFlow機器學習框架搭建和範例原始碼解讀,TensorFlow作為最火熱的機器學習框架之一,Docker是的容器,可以很好的結合起來,為機器學習或者科研人員提供便捷的機器學習開發環境,探索人工智慧的奧秘,容器隨開隨用方便快捷。原始碼解析TensorFlow容器建立和範例程式執行,為熱愛機器學者降低學習難度。

預設機器已經裝好了Docker(Docker安裝和使用可以看我另一篇博文: Ubuntu 16.04安裝Docker1.12+開發範例+hello world+web應用容器 )。

1.下載TensorFlow映象

docker pull tensorflow/tensorflow
#或者
#sudo docker pull tensorflow/tensorflow

2.建立TensorFlow容器,原始碼解讀

docker run --name xiaolei-tensortflow -it -p 8888:8888 -v ~/tensorflow:/notebooks/data  tensorflow/tensorflow
  • docker run 執行映象,
  • --name 為容器建立別名,
  • -it 保留命令列執行,
  • -p 8888:8888 將原生的8888埠 http://localhost:8888/ 對映,
  • -v ~/tensorflow:/notebooks/data 將原生的~/tensorflow資料夾掛載到新建容器的/notebooks/data下(這樣建立的檔案可以儲存到本地~/tensorflow)
  • tensorflow/tensorflow 為指定的映象,預設標籤為latest(即tensorflow/tensorflow:latest)

3.開啟TensorFlow容器

3.1.可以直接從命令列中右鍵開啟連線,或者在瀏覽器中輸入localhost:8888,然後將命令列中的token貼上上去。

4.開始TensorFlow程式設計(Python語言)

4.1.在首頁可以 New 一個Python專案

4.2.tensorflow範例原始碼解讀

from __future__ import print_function
#匯入tensorflow
import tensorflow as tf
#輸入兩個陣列,input1和input2然後相加,輸出結果
with tf.Session():
    input1 = tf.constant([1.0, 1.0, 1.0, 1.0])
    input2 = tf.constant([2.0, 2.0, 2.0, 2.0])
    output = tf.add(input1, input2)
    result = output.eval()
    print("result: ", result)

4.3.執行程式,輸出的結果為(執行成功)

result:  [ 3.  3.  3.  3.]

5.其他 linux,TensorFlow,Docker相關操作

5.1.關閉TensorFlow和開啟TensorFlow環境

#關閉tensorflow容器
docker stop xiaolei-tensortflow

#開啟TensorFlow容器
docker start xiaolei-tensortflow
#瀏覽器中輸入 http://localhost:8888/

5.2.解決檔案的讀寫許可權

#檢視讀寫許可權
ls -l
#將tensorflow 變為屬於xiaolei(系統預設)使用者
sudo chown -R xiaolei tensorflow/
#將tensorflow 變為屬於xiaolei(系統預設)使用者組
sudo chgrp -R xiaolei tensorflow/

更多Docker相關教學見以下內容: 

Docker安裝應用(CentOS 6.5_x64) http://www.linuxidc.com/Linux/2014-07/104595.htm 

Ubuntu 14.04安裝Docker  http://www.linuxidc.com/linux/2014-08/105656.htm 

Ubuntu使用VNC執行基於Docker的桌面系統  http://www.linuxidc.com/Linux/2015-08/121170.htm 

阿里雲CentOS 6.5 模板上安裝 Docker http://www.linuxidc.com/Linux/2014-11/109107.htm 

Ubuntu 15.04下安裝Docker  http://www.linuxidc.com/Linux/2015-07/120444.htm 

在Ubuntu Trusty 14.04 (LTS) (64-bit)安裝Docker http://www.linuxidc.com/Linux/2014-10/108184.htm 

在 Ubuntu 15.04 上如何安裝Docker及基本用法 http://www.linuxidc.com/Linux/2015-09/122885.htm 


IT145.com E-mail:sddin#qq.com