Win下搭建tensorflow的GPU版本

2021-07-25 09:56:30 字數 1201 閱讀 2990

1

python3.5.2

版本並安裝(必須是

3.5版本,而且

3.5後不帶字母的版本,本人親自踩坑!)2、

tensorflow

的gpu

版本,,以及tensorflow

的依賴protubuf

,還有其它比如

numpy

,scipy

等,cuda

是cuda8.0的44

版本3、安裝

cuda8.0

,如果有報警資訊也是可以裝的(如截圖),(印象中我猶豫了很久,最後是抱著試一下的態度裝的),

安裝完後看環境變數裡有以下兩項就表明

cuda

安裝成功!

4cuda

相匹配的

cudnn

,解壓後(如下圖)在環境變數的

path

新增解壓後的

cudnn

的bin

目錄位置變數

5、到這裡基本已經完成

tensorflow

的安裝配置了,如果想驗證一下

cudnn

安裝成功,可以建立乙個

vs的控制台專案,配置平台為

x64,

release

版本(不是

debug

哦),然後在專案屬性—

>vc++

目錄—>

包含目錄加:

d:\cudnn-8.0-windows10-x64-v5.1\cuda\include

(你cudnn

的include

目錄),庫目錄加:

d:\cudnn-8.0-windows10-x64-v5.1\cuda\lib\x64

(你cudnn

的lib

目錄),

鏈結器—

>

附加依賴項新增:

cudnn.lib

。新增如下**:

效果輸出如果是:

cudnn_status_success

,表明cudnn

安裝成功!

6、測試乙個

tensorflow

版本的mnist

程式,在

cmd命令裡執行,有輸出結果如下:

那恭喜你,表明

gpu版本的

tensorflow

安裝成功!

win下通過pip安裝TensorFlow

官方介紹 超詳細 系統環境 win10 一 安裝python開發環境 安裝python3.6 3.4 3.5 安裝pip 安裝 virtualenv pip install virtualenv 安裝 microsoft visual c 2015 redistributable 更新 3 二 建立...

win下通過pip安裝TensorFlow

官方介紹 超詳細 系統環境 win10 一 安裝python開發環境 安裝python3.6 3.4 3.5 安裝pip 安裝 virtualenv pip install virtualenv 安裝 microsoft visual c 2015 redistributable 更新 3 二 建立...

Ubuntu下如何安裝TensorFlow

本文目錄 利用anaconda安裝tensorflow 總結利用docker安裝tensorflow 安裝過程中的一些小問題記錄 tensorflow tensor 向量 flow 流 張量在網路圖中流動 主要參考 1 tensorflow官網tutorials 2 極客學院對tensorflow官...