2021-05-12 14:32:11
基於Docker的TensorFlow機器學習框架搭建和範例原始碼解讀
概述:基於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
相關文章