1. 程式人生 > 程式設計 >tensorflow -gpu安裝方法(不用自己裝cuda,cdnn)

tensorflow -gpu安裝方法(不用自己裝cuda,cdnn)

TensorFlow™是一個基於資料流程式設計(dataflow programming)的符號數學系統,被廣泛應用於各類機器學習(machine learning)演算法的程式設計實現,其前身是谷歌的神經網路演算法庫DistBelief [1] 。

Tensorflow擁有多層級結構,可部署於各類伺服器、PC終端和網頁並支援GPU和TPU高效能數值計算,被廣泛應用於谷歌內部的產品開發和各領域的科學研究 [1-2] 。

tensorflow -gpu安裝

首先,安裝Anoconda

1. 官網下載點我:

2.安裝

點選 python 3.6 version自動下載x64版,下載好之後,然後安裝。

如圖,打上勾之後,一路next

3.開啟終端

1)輸入conda –version 檢視版本

2)配置Python環境

我裝的是python3.5,你們可以根據情況自己選擇

conda create –n tensorflow python=3.5

3)啟用python環境:activate tensorflow

你可以退出當前環境:deactivate tensorflow

劃重點了:安裝tensorflow

**傳統的方法就是用pip 安裝,導致後面還要下載英偉達的cuda支援以及cdnn庫**
**我在這裡介紹一種簡單的方法,不依靠pip**

在終端中輸入

conda install tensorflow-gpu

2.看包中內容

如圖,cuda 9.0和cdnn 7.1.4已經自動安裝了不用再自己下載

3.安裝完之後

開啟Anaconda Prompt 測試成功

總結

以上所述是小編給大家介紹的tensorflow -gpu安裝,最簡單的途徑(不用自己裝cuda,cdnn),希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回覆大家的。在此也非常感謝大家對我們網站的支援!
如果你覺得本文對你有幫助,歡迎轉載,煩請註明出處,謝謝!