小叮噹機器學習:Python3.6配置TensorFlow的GPU版詳細安裝教程

2021-01-04 IT小叮噹

隨著機器學習,神經網絡的搭建,我們對海量數據的處理需求越來越大。

雖然,CPU的發展速度已經放緩,但並行處理的架構卻取得了爆炸式的發展。圖形處理單元(GPU)的大量運用,已經拉開了機器學習迅猛發展的序幕。

谷歌的開源機器學習工具TensorFlow,自發布以來,便迅速躋身於激動人心的機器學習庫的行列之中。

然而網上的大多數都是圍繞python2.7版本的tensorflow教程,下面我們就來看看在Centos7 +pyton3.6的環境下,我們怎麼安裝使用這個功能強大的開源庫的GPU版本。

Step1.環境確認

想要使用GPU版的TesnorFlow來加速我們的神經網絡運算,首先要確保,我們的GPU依賴環境已經搭好。

在終端輸入命令:cat /usr/local/cuda/version.txt (查看cuda版本)

在終端輸入命令: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2(查看cudnn版本)

可以看到這裡cuda版本是8.0.61,而cudnn的版本是5.1.5.(如果沒有安裝cuda和cudnn,到官網根自己的GPU型號版本安裝即可)cuda和cudnn是使用GPU必裝的依賴環境,當然在安裝TensorFlow的GPU版本時,也必不可少。

Step2.安裝Anaconda

在確保,cuda和cudnn已經安裝的GPU可用環境下,我們來使用Anaconda安裝TensorFlow的GPU版。

之所以強烈建議通過Anaconda來安裝TensorFlow,是因為Anaconda集成了較多python第三方庫,對我們後期的神經網絡構建等是十分方便敞亮的。

首先進入官方下載地址:https://www.anaconda.com/download/

由於是Centos7,我們選擇linux版本,之後根據自己安裝的python版本來選擇相應的安裝包版本。小叮噹這裡選擇的是Python3.6版本。

下載完成後,在下載目錄下打開終端,執行命令:bash Anaconda3-5.2.0-Linux-x86_64.sh 根據提示,輸入回車。

之後根據提示輸入"yes",回車確認安裝路徑即可。

注意,此時會提示是否添加路徑,我們輸入「yes",則可以看到系統提示我們,在新打開的終端路徑將被激活。

最後,會提示是否安裝Microsoft vscode,我們可根據個人的需求來輸入yes或no(一般都會選擇no),此時我們輸入命令:conda 卻發現系統提示我們,未找到命令。

我們新打開一個終端,激活路徑,再次輸入命令:conda 發現此時conda已可使用。這表明Anaconda已安裝成功。

Step3.通過Anaconda安裝TensorFlow的GPU版

我們通過在Anaconda下創建虛擬環境來安裝TesnorFlow的GPU版。

(1)首先,我們輸入命令:conda create -n tensorflow python=3.6.5創建一個名叫」tensorflow"的虛擬環境。

此時,系統會提示我們,需要安裝一些新的依賴包,我們輸入"y"即可。

(2)輸入命令:source activate tensorflow 激活環境,之後輸入命令:pip install tensorflow-gpu 安裝Tensorflow的GPU版。

值得注意:此時下載速度極慢,只有83KB/s,於是我們使用命令:pip install tensorflow-gpu -i https://pypi.doubanio.com/simple/ 從豆瓣鏡像中下載,發現速度為11.9MB/s。

不一會兒,TensorFlow的gpu最新版本1.8.0便可成功安裝。

Step4.TensorFlow的GPU版測試

我們輸入命令:python 進入python環境後,輸入命令:import tensorflow 觀察是否報錯。

可以看到出現「ImportError:libculbas.so.9.0:cannot open shared object file:No such file or directory",這是因為我們安裝的tensorflow版本與cuda版本不匹配導致的。

注意:tensorflow1.0版本以上是不支持cuda8.0以下的,何況我們安裝的是最新版得tensorflow-gpu 1.8.0版本。

由於我們在Step1環境確認時,已經可以查看到我們的cuda版本是8.0.61,所以我們選擇低版本的tensorflow -gpu 1.2版本即可。

(1)使用命令:pip uninstall tensorflow-gpu 卸載掉高版本的tensorflow-gpu 1.8.0

(2)安裝上cuda8.0對應的正確tensorflow-gpu版本1.2,使用命令:pip install tensorflow-gpu==1.2 -i https://pypi.doubanio.com/simple/

(3)輸入命令:python之後,輸入:import tensorflow 再次測試。

可以發現,此時我們導入tensorflow已經不會報錯。我們輸入命令:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow! I am IT Xiao Ding Dang!')

sess = tf.Session()

可以看到,當輸入」sess = tf.Session()「後,系統為我們列出了詳細的信息,可以看到此時本機的GPU顯卡GeForce GTX 1070」已經被啟用。這時輸入命令:sess.run(hello)

此時系統已經可以通過tensorflow-gpu為我們正常輸出,這說明我們的tensorflow-gpu已成功安裝。

如果,你還是覺得不放心的話,可以使用這幾句代碼:

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

當你看到gpu時,就說明我們的TensorFlow的GPU版本已經成功安裝!快去愉快的使用吧!

相關焦點

  • 深度學習篇——Tensorflow-GPU配置
    tensoflow-gpu安裝對於python 3.5和3.6的童鞋們而言,安裝tensorflow其實並不難,因為我們可以通過pip直接安裝。不過,第一要求你安裝的python是64位的,如下圖所示,注意劃黃色線的部分。
  • 手把手教你安裝Tensorflow GPU版本
    tensorflow-gpu+電腦win10+gtx1050ti顯卡配置心得       這裡建議使用CUDA8.0的版本,因為小編之前下載安裝了一個CUDA9.0的版本,好像是不能用的哦。之後打開Anaconda Prompt      檢查目前有哪些版本的python可以安裝:conda search --full-name python       創建一個名為tensorflow-gpu的anaconda環境,配置python3.5,之所以選擇3.5的,是因為看的一些博主也是用3.5的,其實也可以用3.6的,因為有對應的tensorflow的版本,小編也是之後才知道的
  • 技術乾貨|tensorflow-gpu 1.4詳細安裝教程
    ://github.com/tensorflow/tensorflow/releases  2.  1.確定版本  TensorFlow GitHub 頁面查看依賴的版本:  https://github.com/tensorflow/tensorflow/releases  2.下載  下載地址:  https://developer.nvidia.com/cudnn  3.安裝
  • 全網最詳細 | Windows 安裝 TensorFlow2.0 GPU 詳細教程
    如果2019的下半年開始入坑TF,那麼你將選擇進入AI的最佳時機了,Tensorflow社區蓬勃發展,未來可期。接下來就給大家帶來TF 2.0 Beta –Window系統下GPU版本的詳細完整的安裝教程。
  • 一文上手最新Tensorflow2.0系列|TensorFlow2.0安裝
    Anaconda官網有各個平臺詳細的安裝使用教程:https://docs.anaconda.com/anaconda/install/。1.提示Anaconda將要安裝的位置,回車 確認。3. 提示是否要寫入配置文件,輸入「yes」回車。4. 提示是否安裝「VSCode」,輸入「no」回車。安裝完成。安裝好後我們鍵入「python3」,此時運行的還是系統自帶的python版本,我們執行「source ~/.bashrc」讓配置生效,此時在執行「python3」,運行的就是Anaconda。
  • 人工智慧神器TensorFlow環境安裝
    Python版本隔離—virtualenv如果我們要同時開發多個應用程式,那這些應用程式都會共用一個Python,就是安裝在系統的Python 3。如果應用A需要python2.7,而應用B需要python3怎麼辦?這種情況下,每個應用可能需要各自擁有一套「獨立」的Python運行環境。
  • TensorFlow 1.9迎來優化版,詳細性能更新乾貨分享!
    Python* Packages (whls)  有三個可用的python packages:  若要安裝英特爾Tensorflow優化版,請運行下面的一個指令將wheel安裝到現有的Python* installation中,最好是Intel® Distribution for Python*:  #Python 2.7  Pip install  https://storage.googleapis.com/intel-optimized-tensorflow/tensorflow-1.9.0
  • tensorflow2.1對應 GPU版本的安裝與卸載
    GTX965M顯卡,win10系統一、檢查tensorflow的Version假定tensorflow 安裝完成,安裝 tensorflow 請參考 集成環境,安裝超簡單>通過 命令行pythonimport tensorflow as tftf.
  • 最新 Tensorflow 2.3 極簡安裝,五分鐘搞定
    2020年8月初,tensorflow 2.3版本發布,Tensorflow的安裝一直是困擾初學者的一個坎,下面我們來看一下如何用五分鐘安裝最新版本的Tensorflow 2.3 。第一步,要檢查和搭建環境。
  • 最新版tensorflow 2.2 的最簡潔安裝
    目前,tensorfow 2.2 正式版已經發布,我們簡單來看一下, tensorflow 2.2 最新版的安裝。首先要明確你希望安裝CPU版本還是GPU版本?安裝GPU版本需要你有nvidia的算力大於3.5的顯卡。
  • 基於RTX2060構建TensorFlow-gpu(keras)學習平臺
    一、conda命令1. onda env list 或 conda info -e 查看當前存在哪些虛擬環境2. conda create --name tf36gpu python=3.6 anaconda3. conda remove -n your_env_name(虛擬環境名稱) --all 刪除虛擬環境4.
  • Win10安裝GPU版本的Tensorflow 2.1
    日期:2020年10月17日首先,下載Anaconda個人版,最新版本是3.7,安裝的時候推薦把Anaconda添加到系統環境變量中。安裝目錄選擇默認配置就好,點下一步等等就安裝完成了。下載安裝,cuda 10.1對應的cudnn 7.6版本。
  • python環境配置簡要流程
    然後再使用 conda create -n tf-gpu python=3.6(版本號) 創建虛擬環境tf-gpu 7、安裝tensorflow 然後選擇 activate tf-gpu
  • 用GPU加速深度學習: Windows安裝CUDA+TensorFlow教程
    使用教程時建議在電腦端閱讀,可以放大圖片。準備步驟請按照介紹下載以下軟體備用,具體安裝流程在第二部分介紹。簡單的說,我們需要 Windows 7/8/10, Server 2012/2016 + Python 3.5 + Visual Studio 2015 + CUDA 8.0 + CuDnn 6.0。
  • win10換了1080ti卡後,安裝tensorflow
    ……扯遠了,這篇文章就講一下啊,這個win10的換了1080ti呢,怎麼一路安到tensorflow。首先,老版本的win10是不支持1080ti的cuda的,他就是不支持,你不樂意也沒辦法。然後……安python不用教了吧……方便小白還是說一下吧。在這Download Python :https://www.python.org/downloads/ 直接下3.X版本,因為windows上張量流不支持2.x版本。一路下一步就好。Anaconda也是一樣,下載了直接安就是了。Download Anaconda Now!
  • 基於Tensorflow\keras銀行卡識別
    本文轉載自【微信公眾號:機器學習算法與Python精研 ,ID:AITop100】,經微信公眾號授權轉載,如需轉載原文作者聯繫來自:GitHub測試環境Ubuntu18.04python 3.6.7numpy 1.16.4tensorflow-gpu 1.13.1 或者是cpu版本keras 2.2.4opencv-python 4.1.0.25PyQt5 5.12.2CUDA
  • tensorflow安裝教程
    tensorflow是谷歌開源的人工智慧庫,有最完善的生態支持。是進行人工智慧領域開發和科研的必備工具。本文在windows10下,藉助anacondaAnaconda安裝和使用,AkShare入門,安裝tensorflow2.0。
  • TensorFlow對象檢測:訓練,導出,優化,推斷
    意思是,除非採取適當措施,否則在具有7.0版計算能力的GPU(V100 Nvidia GPU)上優化的模型不能在具有5.3版計算能力的GPU(Jetson納米板)上運行。https://github.com/tensorflow/tensorflow.git$ cd tensorflow$ git checkout v1.15.2配置TensorFlow# Configure tensorflow$ .
  • Anaconda及tensorflow、pytorch安裝記
    我的python版本3.5.2,有點舊了,跟不上時代,但是更新的話又得卸載再安裝新的版本,這樣一來我原來安裝python的很多庫就得重新安裝了。據說Anaconda這個軟體能解決這個問題,並能很好在安裝時的提供依賴,是安裝順利完成,索性把舊的python卸載了,裝這個試試。
  • Pytorch-GPU1.7.1 和 TensorFlow-GPU1.14.0 安裝指南(基於windows10)
    python=3.7conda activate tensorflow_gpu接著使用如下命令安裝GPU版本的TensorFlow,這種方式可以自動安裝與之匹配的CUDA和CuDNN。conda install tensorflow_gpu=1.14.0