1. 程式人生 > >【深度學習】CentOS 7 安裝GPU版Tensorflow教程(一)

【深度學習】CentOS 7 安裝GPU版Tensorflow教程(一)

之前一直在玩cpu版的tensorflow,這些天突然心血來潮,想搞個gpu版的tensorflow來嚐嚐鮮,沒想到把所有能夠踩的坑幾乎全部踩了一遍,在這裡把自己踩的坑和一些安裝細節拿出來分享給大家,以便讓各位同道中人不要把太多的時間浪費在安裝這件看似很easy的事情上來。


安裝流程:
1.關閉UEFI
       在BIOS裡面檢查你的UEFI是否開啟,如果開啟的話請立馬關掉它(這個很溼重要,因為它很有可能導致你的kernel安裝失敗,筆者就遇到了這個坑,浪費了好多時間),具體怎麼關掉就不多說,由於每種電腦型號的BIOS都有所不同。


2.確認自己的顯示卡支援cuda

[[email protected]
~]$ lspci | grep -i nvidia 01:00.0 VGA compatible controller: NVIDIA Corporation GM107 [GeForce GTX 745] (rev a2) 01:00.1 Audio device: NVIDIA Corporation Device 0fbc (rev a1)

如果有以上資訊出現,說明你的顯示卡是支援cuda的。

3.確認Linux版本是否支援cuda

[[email protected] ~]$ uname -m && cat /etc/*release

若有資訊輸出,說明是支援的。

4.檢查gcc是否安裝

[[email protected] ~]$ gcc --version
gcc (GCC) 4.8.5 20150623 (Red Hat 4.8.5-16)
Copyright (C) 2015 Free Software Foundation, Inc.
This is free software; see the source for copying conditions.  There is NO
warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

若出現以上資訊說明gcc已經安裝。

若沒有安裝,可以使用一下命令安裝
[[email protected] ~]$ sudo yum install gcc gcc-c++

5.安裝kernel-devel和kernel-headers

$ sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
其中$(uname -r)就是個引數,uname -r表示kernel的版本號


6.關閉X server
$ systemctl stop gdm.service


7.禁用nouveau(因為它是一般linxu系統自帶的顯示卡驅動,會和nvidia衝突,所以必須要關掉)
(1)將 nouveau 驅動加入黑名單:
在 /usr/lib/modprobe.d/dist-blacklist.conf 中加入 blacklist nouveau(這種方式僅限在centos 7,其他Linux 系統自行解決)。
(2)備份 initramfs 檔案:

$ sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
(3)重建 initramfs 檔案
$ sudo dracut -v /boot/initramfs-$(uname -r).img $(uname -r)

8.關機重啟

剩餘步驟請檢視CentOS 7 安裝GPU版Tensorflow教程(二)

相關推薦

深度學習CentOS 7 安裝GPUTensorflow教程

之前一直在玩cpu版的tensorflow,這些天突然心血來潮,想搞個gpu版的tensorflow來嚐嚐鮮,沒想到把所有能夠踩的坑幾乎全部踩了一遍,在這裡把自己踩的坑和一些安裝細節拿出來分享給大家,

深度學習GAN生成對抗網路原理詳解1

一個 GAN 框架,最少(但不限於)擁有兩個組成部分,一個是生成模型 G,一個是判別模型 D。在訓練過程中,會把生成模型生成的樣本和真實樣本隨機地傳送一張(或者一個 batch)給判別模型 D。判別模型 D 的目標是儘可能正確地識別出真實樣本(輸出為“真”,或者1),和儘可能

深度學習RNN中梯度消失的解決方案LSTM

上個部落格裡闡述了梯度消失的原因,同時梯度消失會造成RNN的長時記憶失效。所以在本部落格中,會闡述梯度消失的解決方案:①梯度裁剪(Clipping Gradient)②LSTM(Long Short-T

深度學習系列:win10安裝GPUcaffe(python3.6介面

2017/2/14更新: 本人在使用Python3.6安裝成功後任然存在一些問題,很麻煩。我也去仔細看了下caffe支援的版本,目前只支援到3.5,所以大家可以參考我的另外一篇部落格安裝caffe。本來想直接刪除這篇部落格,但對於一些還不知情的朋友,希望能提醒

深度學習谷歌雲GPU伺服器建立與使用指南

sudo wget -O driver-384.deb http://cn.download.nvidia.com/tesla/384.66/nvidia-diag-driver-local-repo-ubuntu1604-384.66_1.0-1_amd64.deb按照官網“其他”中介紹的安裝方式安裝  2

深度學習谷歌deepdream原理及tensorflow實現

什麼是DeepDream? DeepDream是谷歌釋出的對卷積神經網路(CNN)進行視覺化的方法,當然它的用途不僅限於此,我們可以通過它讓機器“做夢”,以下是一些效果: 可以看到計算機將自然影象的

Fiddler學習Fiddler簡介和Web抓包應用

  一、Fiddler是什麼?     Fiddler是一個http協議除錯代理工具,它能夠記錄並檢查所有你的電腦和網際網路之間的http通訊,設定斷點,檢視所有的進出Fiddler的資料。 Fiddler 要比其他的網路偵錯程式要更加簡單,因為它不僅僅暴露http通

機器學習使用Python的自然語言工具包NLTK對Reddit新聞標題進行情感分析

讓我們使用Reddit API獲取新聞標題並執行情感分析 在我上一篇文章中,使用Python進行K-Means聚類,我們只是抓取了一些預編譯資料,但是對於這篇文章,我想更深入地瞭解一些實時資料。 使用Reddit API,我們可以從各種新聞subreddit獲得成千上萬的

備忘 Linux系統安裝時分割槽的選擇推薦

前言: 以前初識Linux時,對Linux系統安裝時分割槽的選擇,一點都不瞭解,導致幾次沒法進行下一步安裝,因此就靜下心來,專門拿出時間研究了研究這方面的知識; 以下內容就是以前通過研究Linux安裝過程中分割槽的知識總結出來的一些知識;貼上來,做個備份,也希望對Linux初

機器學習動手寫一個全連線神經網路:分類

我們來用python寫一個沒有正則化的分類神經網路。 傳統的分類方法有聚類,LR邏輯迴歸,傳統SVM,LSSVM等。其中LR和svm都是二分類器,可以將多個LR或者svm組合起來,做成多分類器。 多分類神經網路使用softmax+cross entropy組

機器學習利用蟻群演算法求解旅行商TSP問題

如果喜歡這裡的內容,你能夠給我最大的幫助就是轉發,告訴你的朋友,鼓勵他們一起來學習。 If you like the content here, you can give me the greatest help is forwarding, tell you

集合框架JDK1.8源碼分析之HashMap 轉載

.get 修改 object set implement .com 功能 數組元素 帶來 一、前言   在分析jdk1.8後的HashMap源碼時,發現網上好多分析都是基於之前的jdk,而Java8的HashMap對之前做了較大的優化,其中最重要的一個優化就是桶中

Android 動畫動畫詳解之補間動畫

前言 :之前很早就想寫寫Android 的動畫,最近剛好有時間,大概聊一聊安卓動畫。 我個人習慣將動畫分為:補間動畫(透明度、旋轉、位移、縮放)、幀動畫、和屬性動畫,這一篇,我們先說說補間動畫。 補間動畫這個詞出於flash,在兩個關鍵幀(可以理解成動畫開始和結束)中間需要做“補

Android架構基於MVP模式的Retrofit2+RXjava封裝

#最近有個新專案要做,搭建框架的時候,順便梳理了下MVP模式,特此記錄,歡迎大家指正。 專案地址GitHub 一 、首先是依賴 compile 'com.google.code.gson:gson:2.8.0' compile 'com.squareup.ok

吳恩達深度學習系列課程筆記:卷積神經網路

本系列文章將對吳恩達在網易公開課“深度學習工程師”微專業內容進行筆記總結,這一部分介紹的是“卷積神經網路”部分。 1、計算機視覺 計算機視覺在我們還是生活中有非常廣泛的應用,以下幾個是最常見的例子: 影象分類: 可以對影象中的物體種類進行判斷,如確定影象中

SSH實戰IntelliJ IDEA環境開發BOS物流專案環境搭建

一、專案概述二、搭建專案開發環境(一)資料庫環境/*建立一個數據庫*/ CREATE DATABASE bos CHARACTER SET utf8; /*建立一個新使用者*/ CREATE USER lee IDENTIFIED BY 'root'; /*對新使用者進行授權

深度學習之卷積神經網路原理詳解

初探CNN卷積神經網路 1、概述 典型的深度學習模型就是很深層的神經網路,包含多個隱含層,多隱層的神經網路很難直接使用BP演算法進行直接訓練,因為反向傳播誤差時往往會發散,很難收斂 CNN節省訓練開銷的方式是權共享weight sharing,讓一組神經元

深度學習結合SLAM的研究思路/成果整理之使用深度學習方法替換SLAM中的模組

整理了部分近兩年深度學習結合SLAM的一些研究成果(參考知乎帖子https://www.zhihu.com/question/66006923 和泡泡機器人公眾號,附上論文連結和已找到的原始碼/資料集連結,大多簡單看了一下摘要。僅為自己學習所用,確實翻譯得很爛…

深度學習應用系列——計算機視覺:表面缺陷檢測

Fully Convolutional Networks for Surface Defect Inspection in Industrial Environment 佔坑 一種思路:使用語義分割

純感悟我眼裡的安卓架構師

       提起架構師,大家會覺得很神祕很高深,但是真的如此麼?安卓架構師是在現實的產品需求基礎上對技術的一些整體構思,要考慮系統性,實用性,優化性,邏輯性,可擴充套件性,要滿足產品各方面的技術需求。相當於是頂層設計,要兼顧到各個方面。下面就具體的說一下: