1. 程式人生 > >Ubuntu16.04安裝CUDA+cuDNN+GPU版TensorFlow過程記錄

Ubuntu16.04安裝CUDA+cuDNN+GPU版TensorFlow過程記錄

一、安裝環境說明

軟體 版本 下載連結 說明
Ubuntu系統 16.04-64位
Python anaconda 3.6 不使用系統自帶的2.7版本
CUDA cuda_8.0.61_375.26_linux.run NVIDIA推出的使用GPU資源進行通用計算的SDK,CUDA的安裝包裡一般集成了顯示卡驅動
cuDNN cudnn-8.0-linux-x64-v5.1.tgz cuDNN(需要註冊帳號,這裡提供該版本的百度網盤下載 cuDNN是NVIDIA推出的深度學習中CNN和RNN的高度優化實現
TensorFlow tensorflow_gpu_1.0.1

二、安轉過程快速瀏覽

1.安裝前準備

1.1 Verify You Have a CUDA-Capable GPU(確認你有一個可用CUDA的GPU)

$ lspci | grep -i nvidia

這裡寫圖片描述
可以看到,我有三塊可用的GeForce GTX 1080,你可以在這裡檢視你的GPU是否是CUDA-caplable。

1.2 Verify You Have a Supported Version of Linux(確認你的Linux版本支援CUDA)

CUDA開發工具只支援幾個特定的Linux發行版本
這裡寫圖片描述

#檢視你的系統資訊
$ uname -m && cat /etc/*release

這裡寫圖片描述

1.3 Verify the System Has gcc Installed(查證你是否安裝了gcc)

$ gcc --version

這裡寫圖片描述

1.4 Verify the System has the Correct Kernel Headers and Development Packages Installed(驗證kernel headers和development packages)

# 檢視核
$ uname -r
# 為當前核安裝kernel headers和development packages
$ sudo apt-get install linux-headers-$(uname -r)

2. 選擇一種安裝方式(runfile installation)

使用cuda_8.0.61_375.26_linux.run安裝

2.1 確認之前的準備工作已經完成

2.2 Disable the Nouveau drivers(禁用nouveau,這一步至關重要!!!)

2.2.1使用以下命令檢視Noueau驅動是否被載入了

$ lsmod | grep nouveau
# 如果打印出一些資訊,說明Noueau被載入了,正常情況下會打印出一些關於nouveau的資訊

2.2.2 Create a file at /etc/modprobe.d/blacklist-nouveau.conf with the following contents:
建立/etc/modprobe.d/blacklist-nouveau.conf,寫入:

blacklist nouveau
options nouveau modeset=0

這裡寫圖片描述
這裡寫圖片描述
2.2.3 Regenerate the kernel initramfs:

$ sudo update-initramfs -u

3.Reboot into text mode (runlevel 3)(重啟,進入文字模式)

重啟後按: Ctrl+Alt+F1 進入文字模式
這裡寫圖片描述

3.1 確認你的Nouveau沒有被載入

Verify that the Nouveau drivers are not loaded. If the Nouveau drivers are still loaded, consult your distribution’s documentation to see if further steps are needed to disable Nouveau.
如何確認,請看 2.2.1
停止X服務

$ sudo /etc/init.d/lightdm stop

3.2 執行可執行檔案

sudo sh cuda_8.0.61_375.26_linux.run

這裡寫圖片描述

簡單一點,除了Install the CUDA 8.0 Samples? (y)es/(n)o/(q)uit選擇n外,其他資訊都是預設和y.按n鍵不選擇安裝CUDA samples,我們只通過TensorFlow呼叫CUDA,不直接寫CUDA程式碼,如果需要請選擇’y’自行開啟。
最後等待安裝程式完畢。

3.3 安裝完成後

check device nodes (檢查裝置節點): Check if /dev/nvidia* files exist。如果沒有的話$ sudo modprobe nvidia

重啟桌面服務

$ sudo service lightdm start

此時能夠登入,沒有login-loop問題。

4. 在系統環境裡設定CUDA的路徑

$ vim ~/.bashrc # 編輯

~/.bashrc 的最後中新增如下程式碼

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-8.0
export PATH=/usr/local/cuda-8.0/bin:$PATH
$source ~/.bashrc #配置立即生效

5. 安裝cuDNN

cd到cuda的安裝目錄執行解壓命令:

$ cd /usr/local
$ sudo tar -xzvf ~/downloads/cudnn-8.0-linux-x64-v5.1.tgz

這樣就完成了cuDNN的安裝。

6. 驗證驅動版本和CUDA版本:

$ cat /proc/driver/nvidia/version 
$ nvcc -V

三、安裝tensorflow

這裡寫圖片描述

這裡寫圖片描述

這裡寫圖片描述

安裝TensorFlow並測試,成功!!!

四、在jupyter notebook中新增tensorflow kernel

# python3 -m venv .env
# source .env/bin/activate
(.env)# python -m pip install ipykernel
(.env)# python -m ipykernel install --user --name my_project --display-name "My Project"

這裡寫圖片描述
啟動jupyter notebook
這裡寫圖片描述
這裡寫圖片描述
建立一個tensorflow kernel的test_tf.ipynb
這裡寫圖片描述

如果實在遠端主機上安裝的,想要通過ip地址在本地訪問jupyter notebook,請參考這篇部落格

以上就是所有的安裝過程,接下來就是要學習怎麼使用TensorFlow了,加油!!!

相關推薦

Ubuntu16.04安裝CUDA+cuDNN+GPUTensorFlow過程記錄

一、安裝環境說明 軟體 版本 下載連結 說明 Ubuntu系統 16.04-64位 … … Python anaconda 3.6 不使用系統自帶的2.7版本 CUDA cuda_8

ubantu安裝cudaGPUtensorflow(詳細)

首先十分感謝我的老師對我的支援,終於申請到了電腦,現在終於可以在linux系統下載配置tensorflow-GPU加速器了,對於新手說安裝過程也是十分’感動’,掉坑無數,希望本文可以給大家帶來幫助。 步驟一:安裝Nvidia顯示卡驅動 步驟二:下載並安裝cu

ubuntu16.04 安裝CUDA 8.0 和 cuDNN 5.1 /cudnn6.0,可適用於gpu版本的(tensorflow,caffe,mxnet)

轉載https://zhuanlan.zhihu.com/p/27890924文章,略有修改,感謝原作者 環境: ubuntu 16.04GTX 960 安裝步驟 安裝Nvidia驅動 系統設定 --> 軟體與更新 --> 附加驅動,如下圖選擇nvidia官

Ubuntu16.04 安裝 CUDACUDNN、OpenCV 並用 Anaconda 配置 Tensorflow 和 Caffe 詳細過程

1.配置環境 2.安裝顯示卡驅動、CUDA 和 CUDNN 在Ubuntu系統中的 System Settings -> Software & Updates -> Additional Drivers 安裝相應的顯示卡和C

Ubuntu16.04+anaconda+cuda+cudnn+caffe+opencv新手曲折安裝之路

你還按照caffe官網來裝嗎? 歡迎你入坑了,經過多天探索,無意間發現了寶藏,下面介紹及其簡單的方法(賊簡單,不過anaconda有些軟體安裝不完全,意味著可能跑程式碼時發現此安裝方法不行。。) 首先恭喜看到這篇文章的朋友,如果你經過多天探索無果來到這裡,那麼開啟火箭安裝模式吧,如果你在探索道路

Ubuntu16.04安裝cudacudnn

開啟CUDA 8.0 Samples預設安裝路徑,終端輸入 cd /home/its/NVIDIA_CUDA-8.0_Samples (its是我的使用者名稱) sudo make all –j16 (16核,根據你的電腦配置,核越多執行越快)出現“unsupported GNU version! gcc v

ubuntu16.04 安裝 CUDA

too vim ide html body 關閉 ati ssh 匹配 1. 詳細資料參見 https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#runfile-nouveau 2. 遠程

ubuntu16.04安裝cuda,無法定位軟件包問題

應該 網上 定位 安裝包 clas 裝包 粘貼 source 離線 為了學習深度學習,這幾天在安裝深度學習框架,cuda安裝時卻出現無法定位軟件包的問題。cuda官網下載時提供了deb和run格式,今天只講deb格式的安裝包安裝過程的問題。 按照官方教程,下載好cuda d

重灌Ubuntu 恢復 Cuda + cudnn + GPU driver + tensorflow 環境走過的坑

接上 最近要重現YOLOv3 實時線上物體檢測演算法,程式碼要求Python3.6及以上由於Ubuntu 自帶的Python版本是3.5.2於是索性直接給卸掉了, 由於是Ubuntu自帶的版本, 可能系統需要用到裡面的元件,再次重啟系統的時候產生登陸迴圈login

Ubuntu16.04 +cuda8.0+cudnn+caffe+theano+tensorflow配置明細

歡迎關注我的部落格:http://blog.csdn.net/hit2015spring和http://www.cnblogs.com/xujianqing 本文主要是介紹在ubuntu16.04下,怎麼配置當下流行的深度學習框架,cuda8.0+cudnn+caffe+theano+tensorflow 安

ubuntu 18.04 安裝 cuda-cudnn-NVIDIA 以及美化 全套(小白個人配置過程

我所有的命令全在root許可權下進行的所以先進行登入root使用者 右擊滑鼠開啟終端 (英文是:terminal)   終端更換root使用者 需要使用 sudo -i 若想更改密碼則為 sudo passwd 輸入兩次新密碼 跟換使用者 su ro

ubuntu16.04 安裝caffe(GPU

1、首先安裝caffe之前我已經安裝好基於GPU的tensorflow,所以顯示卡驅動+cuda+cudnn我都已經安裝好。 教程如下: http://blog.csdn.net/xhjj520/article/details/78857353 2、另外 安裝opencv2

Ubunt16.04安裝CUDA cuDNN OpenCV Caffe完整教程

一、 安裝CUDA 版本 cuda_8.0.61_375.26_linux.run 安裝過程 前期準備 1、確認有Nvidia的GPU $ lspci | grep -i nvidia 2、確認你的GPU是否是C

ubuntu16.04 安裝caffe 無GPU

export PYTHONPATH=/home/tom/caffe/python:$PYTHONPATH /home/tom/caffe/python為自己安裝 caffe下的python 資料夾所在的路徑 本人遇到make pycaffe出現錯

ubuntu16.04安裝CUDA 8.0(很詳盡,包括一些坑的解決方法)

每次配置環境都頭疼得要死,總是會遇到各種各樣的問題,這次決心一勞永逸地解決這個問題,踩最多的坑,裝最穩的環境。 先介紹用於測試的電腦:作業系統是在清華映象上下載的ubuntu16.04LTS,核心版本是4.15.0-29-generic,電腦是聯想Y430P,顯示卡我沒記錯的話是GTX850M。 本篇絕大

Ubuntu16.04安裝caffe(CPU版本)過程

具體安裝過程中遇到的幾點問題 1.第三步修改檔案路徑時記得注意空格; 2.錯誤:You are using pip version 9.0.3, however version 10.0.1 is available. You should consider

ubuntu16.04下跑通LSD-SLAM的過程記錄

height pla gen lease list wid 站點 roc pytho 第一步:首先安裝ROS kinetic:   參考:https://blog.csdn.net/softimite_zifeng/article/details/78632211     

Ubuntu16.04安裝配置GPUTensorFlow

基本工作 更新系統 sudo apt-get update sudo apt-get upgrade -y sudo apt-get dist-upgrade -y 安裝linux kernel modules,linux source和headers

Ubuntu16.04CUDA 9.0 + cuDNN v7.0 + tensorflow 1.6.0(GPU)環境搭建

由於自己攢了個主機,第一次安裝GPU版本的tensorflow,mark一下。說一下環境的版本:系統:Ubuntu 16.04.3顯示卡:gtx 1080(索泰 至尊Plus OC)CUDA 9.0cuDNN v7.0.5 for CUDA 9.0tensorflow 1.6

ubuntu16.04安裝Nvidia顯示卡驅動,CUDA8.0,cuDNN6,TensorFlow-gpu

Welcome To My Blog 實驗室學姐讓跑一個深度學習模型的程式,需要配置環境,查閱各種資料後安裝成功,記錄一下安裝過程,確實挺刺激,也算是開啟了自己的深度學習之旅了. 安裝Nvidia顯示卡驅動 1. 下載驅動 根據自己的系統和顯示卡型號直接在NVIDIA官網