1. 程式人生 > >Windows 下安裝 TensorFlow-gpu 教程

Windows 下安裝 TensorFlow-gpu 教程

終於邁開了這一步,TensorFlow 大法好。

一、介紹

TensorFlow™ 是一個採用資料流圖(data flow graphs),用於數值計算的開源軟體庫。節點(Nodes)在圖中表示數學操作,圖中的線(edges)則表示在節點間相互聯絡的多維資料陣列,即張量(tensor)。它靈活的架構讓你可以在多種平臺上展開計算,例如臺式計算機中的一個或多個 CPU(或 GPU),伺服器,移動裝置等等。TensorFlow 最初由 Google 大腦小組(隸屬於 Google 機器智慧研究機構)的研究員和工程師們開發出來,用於機器學習和深度神經網路方面的研究,但這個系統的通用性使其也可廣泛用於其他計算領域。(來自

TensorFlow中文社群
在這裡插入圖片描述

一開始還沒發現這圖示的意思,直到偶然間,發現正好是 TF 的影子重疊在一起構成的。TensorFlowLOGO 是真的好看。

二、安裝 TensorFlow

我是直接用 pip 安裝的 GPU 版本。

pip3 install --upgrade tensorflow-gpu

這個安裝好之後,還不能跑 TensorFlow ,還缺兩個東西。

三、安裝 CUDA

這個你要確定版本,非常重要!否則不能使用。可以先找到對應的顯示卡:網址在這裡。我的是 GTX960M 的顯示卡。

我當時按照上面的指令安裝 TensorFlow 的時候,自動裝的是比較新的版本。然後我就去裝了最新的 CUDA

,結果發現不能用,又試了幾個版本,最終裝了 CUDA 9.0 才成功。

先去官網找到對應的版本。
在這裡插入圖片描述

再選擇相應的資訊,點選 Download 即可。
在這裡插入圖片描述

下載完後,直接點選 exe 進行安裝。下面演示的是 CUDA 10 安裝過程。
在這裡插入圖片描述
在這裡插入圖片描述
在這裡插入圖片描述

選擇 精簡 安裝。

在這裡插入圖片描述

四、安裝 cuDNN

首先,要去官網註冊個賬號。然後,選擇與 CUDA 相對應的版本。
在這裡插入圖片描述

下載完解壓,我是把 cuda 資料夾解壓到 C盤 的。
在這裡插入圖片描述

新增環境變數。
在這裡插入圖片描述

五、測試

import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c =
tf.matmul(a, b) sess = tf.InteractiveSession() sess.run(c)

如果看到執行出下圖的結果,就說明安裝成功了。
在這裡插入圖片描述

六、致謝