如何透過 Container Station 使用 TensorFlow

最後修訂日期: 2019-09-24

關於 TensorFlow

TensorFlow™ 是一款開放原始碼的軟體庫，可透過資料流程圖執行數值演算。它以圖形中的「節點」代表數學運算，而用圖形的「邊」代表聯繫各節點的多維度資料陣列（張量）。

在 Container Station 安裝 TensorFlow

指派 GPU 至 Container Station。
1. 前往〔控制台〕>〔系統〕>〔硬體〕>〔顯示卡〕。
2. 在〔資源使用〕下方，指派 GPU 至〔Container Station〕。
3. 點擊〔套用〕。
開啟〔Container Station〕。

使用正確的映像檔版本。

點擊〔映像檔〕。

選擇已安裝的所需映像檔並點擊〔提取〕。

注意：建議您根據安裝的 QTS 和 Nvidia 驅動程式版本，使用下列版本的 TensorFlow：

QTS 和 Nvidia 驅動程式版本	標籤	提取指令
QTS 4.3.5 和 Nvidia 驅動程式 v1.3.5	tensorflow/tensorflow:1.11.0-gpu	docker pull tensorflow/tensorflow:1.11.0-gpu
QTS 4.4.x 和 Nvidia 驅動程式 v2.0.0	tensorflow/tensorflow:1.11.0-gpu	docker pull tensorflow/tensorflow:1.11.0-gpu

點擊〔建立〕。
搜尋關鍵字〔TensorFlow〕。找到〔tensorflow/tensorflow〕並點擊〔安裝〕。

根據您安裝的 QTS 和 Nvidia 驅動程式版本，選擇特定版本的 TensorFlow。

QTS 和 Nvidia 驅動程式版本	建議版本
QTS 4.3.5 和 Nvidia 驅動程式 v1.3.5	1.11.0-gpu
QTS 4.4.x 和 Nvidia 驅動程式 v2.0.0	1.11.0-gpu

點擊〔下一步〕。
點擊〔進階設定〕。
指派 GPU 至容器。
1. 前往〔裝置〕。
2. 點擊〔新增〕。
3. 選擇要加入容器的 GPU。
非必要步驟： 與容器共享 NAS 資料夾。
1. 前往〔共用資料夾〕。
2. 在〔主機的磁碟區〕上方點擊〔新增〕。
  隨即新增新的主機磁碟區。
3. 選擇主機路徑。
4. 指定掛載點。
點擊〔建立〕。
畫面隨即顯示新容器的〔摘要〕。
檢視容器的設定值。
點擊〔確定〕。
容器映像檔完成安裝。

透過 SSH 掛載 NVIDIA GPU

透過 SSH 連線至 NAS。

掛載 GPU 至容器。

根據您要掛載的 GPU 輸入下列其中一個指令。

要掛載的 GPU

指令

第一個

–device /dev/nvidia0:/dev/nvidia0 \
–device /dev/nvidiactl:/dev/nvidiactl \
–device /dev/nvidia-uvm:/dev/nvidia-uvm \
-v `/sbin/getcfg NVIDIA_GPU_DRV Install_Path -f
/etc/config/qpkg.conf -d None`/usr/:/usr/local/nvidia

第二個

–device /dev/nvidia0:/dev/nvidia0 \
–device /dev/nvidia1:/dev/nvidia1 \
–device /dev/nvidiactl:/dev/nvidiactl \
–device /dev/nvidia-uvm:/dev/nvidia-uvm \
-v `/sbin/getcfg NVIDIA_GPU_DRV Install_Path -f
/etc/config/qpkg.conf -d None`/usr/:/usr/local/nvidia

兩個都掛載

–device /dev/nvidia1:/dev/nvidia1 \
–device /dev/nvidiactl:/dev/nvidiactl \
–device /dev/nvidia-uvm:/dev/nvidia-uvm \
-v `/sbin/getcfg NVIDIA_GPU_DRV Install_Path -f
/etc/config/qpkg.conf -d None`/usr/:/usr/local/nvidia

注意：不同 QTS 和 Nvidia 驅動程式版本適用的指令範例如下所示：

QTS 和 Nvidia 驅動程式版本

指令

QTS 4.3.5/4.3.6 和 Nvidia 驅動程式 v1.3.5

docker run -d –name tensorflow \
    –device /dev/nvidia0:/dev/nvidia0 \
    –device /dev/nvidiactl:/dev/nvidiactl \
    –device /dev/nvidia-uvm:/dev/nvidia-uvm \
    -v `/sbin/getcfg NVIDIA_GPU_DRV Install_Path -f /etc/config/qpkg.conf -d 
None `/usr/:/usr/local/nvidia \
     -p 6006:6006 -p 8888:8888 \
    tensorflow/tensorflow:1.11.0-gpu

QTS 4.4.x 和 Nvidia 驅動程式 v2.0.0

docker run -d –name tensorflow \
   –device /dev/nvidia0:/dev/nvidia0 \
   –device /dev/nvidiactl:/dev/nvidiactl \
   –device /dev/nvidia-uvm:/dev/nvidia-uvm \-v `/sbin/getcfg NVIDIA_GPU_DRV Install_Path -f 
/etc/config/qpkg.conf -d
None`/usr/:/usr/local/nvidia \
    -p 6006:6006 -p 8888:8888 \
    tensorflow/tensorflow:1.11.0-gpu