如何安裝 TensorFlow (Windows 環(huán)境下)

前置檢查

  • 已安裝 Python 2.5 及以上版本(建議安裝 3 以上版本)
    請(qǐng)自行到官網(wǎng)下載安裝。
  • 已安裝 Numpy
    獲取地址(請(qǐng)根據(jù)自己的系統(tǒng)及需求下載)
    注1:也可以在 cmd 中使用 pip 安裝(前提是已經(jīng)安裝 pip,Python3 及以上版本安裝好以后就自帶的),在命令行中輸入:
    pip3 install numpy
    然后等待安裝成功。
    注2:安裝過程中若出現(xiàn)問題請(qǐng)到 這里 查詢。
  • 已安裝 VisualStudio 2015
    這個(gè)教程很多,隨便百度一下就會(huì)出來很多。

使用 pip 安裝 TensorFlow

  • 安裝 GPU 版 (運(yùn)行速度更快,推薦,但只有 NVIDIA 的顯卡才能安裝)
    cmd 中輸入 pip install tensorflow-gpu
  • 安裝 CPU
    cmd 中輸入 pip install tensorflow

安裝 cuda

劃重點(diǎn)?。。?! 在安裝之前,請(qǐng)務(wù)必關(guān)閉 360安全衛(wèi)士/騰訊管家 等安全軟件?。?!

  • 安裝之前請(qǐng)先到 這里 查詢你的電腦顯卡是否支持 cuda
  • 官網(wǎng) 下載自己需要的版本
    注:建議下載本地版,網(wǎng)絡(luò)版會(huì)從網(wǎng)頁獲取數(shù)據(jù),可能會(huì)失??;
  • 雙擊 .exe 文件開始安裝
  • 如果選擇自定義安裝,建議使用默認(rèn)路徑。
  • 安裝完成后,可在 cmd 中輸入 nvcc -V 來查看當(dāng)前已安裝 cuda 的版本信息,如果輸出了版本信息,說明安裝成功。

下載 cudnn

  • 同理,到官網(wǎng)下載對(duì)應(yīng)你的 cuda 版本的 cudnn。
    注1:如果安裝的是 cuda8 ,請(qǐng)下載 cudnn v6,因?yàn)槭褂?pip 安裝的是最新版的 TensorFlow,而 TensorFlow1.3 就已經(jīng)不再支持 cudnn v5,而支持 cudnn v6。
    注2:下載完成后是一個(gè)壓縮包,解壓后有三個(gè)文件,如圖
    cudnn 解壓后

    請(qǐng)將三個(gè)文件夾一并拷入安裝 cuda 中的 cuda 文件夾中,如:
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8
    ,與命名相同的文件夾合并,如圖:
    cuda 文件夾
  • 檢測(cè)系統(tǒng)環(huán)境變量 PATH 是否配置好,例如:
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp

檢查是否能正常使用

  • 在 菜單->程序->Python 中,找到 Python 自帶的編輯器 IDLE,打開,并輸入:
    import tensorflow as tf
    回車,若成功,則安裝成功。

補(bǔ)充:

  1. 簡單矩陣乘法 測(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)

輸出:

output
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容