1. 程式人生 > >ubuntu16.04下安裝CUDA,cuDNN及tensorflow-gpu

ubuntu16.04下安裝CUDA,cuDNN及tensorflow-gpu

原博文

這篇文章主要依據兩篇文章: 

深度學習主機環境配置: Ubuntu16.04+Nvidia GTX 1080+CUDA8.0 
深度學習主機環境配置: Ubuntu16.04+GeForce GTX 1080+TensorFlow 
不過在實際執行的過程中,有一定的不同之處,隨著時間的推移,一些元件已經可以更方便的安裝,不再需要自己編譯了。一些流程也有所更改。因此我在這裡把自己在ubuntu16.04下安裝cuda,cudnn和gpu版tensorflow的流程寫下來,供人蔘考。

安裝顯示卡驅動

首先安裝顯示卡驅動。首先看自己顯示卡

lspci | grep -i vga
lspci | grep -i nvidia

然後看顯示卡驅動

lsmod | grep -i nvidia

在ubuntu16.04中,更換驅動非常方便,去 
系統設定->軟體更新->附加驅動->切換到最新的NVIDIA驅動即可。應用更改->重啟

選區_052.png-132.8kB

再執行nvidia-smi來看看

選區_053.png-55.8kB

安裝CUDA

sudo ./cuda_8.0.44_linux.run --tmpdir=/tmp

開始安裝。後面引數 tmpdir好像是為了放日誌,具體是什麼日誌忘了=。= 安裝開始以後,首先是一個協議,一直按回車到底以後,輸入accept。其他的操作如下所示

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for
Linux-x86_64 367.48? (y)es/(n)o/(q)uit: n Install the CUDA 8.0 Toolkit? (y)es/(n)o/(q)uit: y Enter Toolkit Location [ default is /usr/local/cuda-8.0 ]: Do you want to install a symbolic link at /usr/local/cuda? (y)es/(n)o/(q)uit: y Install the CUDA 8.0 Samples? (y)es/(n)o/(q)uit: y

下面是安裝時的輸出資訊:

Installing the CUDA Toolkit in
/usr/local/cuda-8.0 … Installing the CUDA Samples in /home/textminer … Copying samples to /home/textminer/NVIDIA_CUDA-8.0_Samples now… Finished copying samples. =========== = Summary = =========== Driver: Not Selected Toolkit: Installed in /usr/local/cuda-8.0 Samples: Installed in /home/textminer Please make sure that – PATH includes /usr/local/cuda-8.0/bin – LD_LIBRARY_PATH includes /usr/local/cuda-8.0/lib64, or, add /usr/local/cuda-8.0/lib64 to /etc/ld.so.conf and run ldconfig as root To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-8.0/bin Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-8.0/doc/pdf for detailed information on setting up CUDA. ***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 361.00 is required for CUDA 8.0 functionality to work. To install the driver using this installer, run the following command, replacing with the name of this run file: sudo .run -silent -driver Logfile is /tmp/cuda_install_6583.log

安裝完畢後,再宣告一下環境變數,並將其寫入到 ~/.bashrc 的尾部:

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

儲存退出,執行source ~/.bashrc 
測試是否安裝成功

// 如果怕把samples搞壞了那就先搞一個備份,在備份裡搞
cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

結果如下

選區_054.png-169.8kB

之前有看到說要把gcc改成4.9的。不過我現在用5.4的也可以編譯,就是有幾個警告,不知道會不會有什麼大的影響。

cuDNN安裝

如果要使用gpu來對tensorflow進行加速,除了安裝CUDA以外,cuDNN也是必須要安裝的。跟cuda一樣,去nvidia的官網下載cuDNN的安裝包。不過這次沒法直接下載,需要先註冊,然後還要做個調查問卷什麼的,稍微有點麻煩。我下的是cuDNN v5.1 Library for Linux這個版本。不要下cuDNN v5.1 Developer Library for Ubuntu16.04 Power8 (Deb)這個版本,因為是給powe8處理器用的,不是amd64.

下載下來以後,發現是一個tgz的壓縮包,使用tar進行解壓

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安裝cuDNN比較簡單,解壓後把相應的檔案拷貝到對應的CUDA目錄下即可

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

tensorflow安裝

我之前已經安裝了cpu-only版的tensorflow,所以現在要先把原先的tf解除安裝

sudo pip uninstall tensorflow
sudo pip install tensorflow_gpu-0.12.0rc1-cp35-cp35m-linux_x86_64.whl

來測試一下

$ipython
import tensorflow as tf

結果如下

選區_055.png-81.5kB

沒有報錯,說明已經安裝成功了。

在idea中呼叫tensorflow

之前在命令列中已經呼叫tensorflow成功了,沒想到在idea中寫程式時呼叫tensorflow又出現了錯誤:

ImportError: libcudart.so.8.0: cannot open shared object file: No such file or directory

Error importing tensorflow.  Unless you are using bazel,
you should not try to import tensorflow from its source directory;
please exit the tensorflow source tree, and relaunch your python interpreter
from there.

選區_056.png-10kB

但是我並沒有在tf的根目錄下執行呀?而且同一個檔案用命令列是可以執行的呀?為什麼換idea來就不行了捏? 
後來經過嘗試發現,問題還是出在環境變數LD_LIBRARY_PATH中。光把環境變數寫在~/.bashrc中是不行的,還需要寫在/etc/profile下。因此,可以在/etc/profile的結尾處加上

## cuda
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

登出後重新登陸,再開啟idea,就可以在idea中呼叫tf啦

選區_057.png-25.9kB

相關推薦

ubuntu16.04安裝CUDAcuDNNtensorflow-gpu版本過程

這篇文章主要依據兩篇文章: 深度學習主機環境配置: Ubuntu16.04+Nvidia GTX 1080+CUDA8.0 深度學習主機環境配置: Ubuntu16.04+GeForce GTX 1080+TensorFlow 不過在實際執行的過程中,

ubuntu16.04安裝CUDAcuDNNtensorflow-gpu

原博文:這篇文章主要依據兩篇文章: 深度學習主機環境配置: Ubuntu16.04+Nvidia GTX 1080+CUDA8.0 深度學習主機環境配置: Ubuntu16.04+GeForce GTX 1080+TensorFlow 不過在實際執行的過程中,有一定的不同之處

ubuntu16.04解除安裝cudacudnn以及檢視版本號

ubuntu 解除安裝cuda 在命令列中輸入 sudo apt-get remove cuda sudo apt-get autoclean sudo apt-get remove cuda* 然後在目錄切換到 cd /usr/local/ ~$:sudo rm -r cu

ubuntu16.04安裝caffe遇到的系列問題純小白犯得錯誤勿噴

—>安裝opencv時 出現cmake error at 3rdparty/ippicv/downloader.cmake:73 (file) 這是因為我的這個ippicv_linux_20141027.tgz安裝包出現的問題, 在opencv-3.1.0

Ubuntu16.04安裝opencv2.4.9若干問題解決方法

1. 下載opencv2.4.9 安裝包 wget https://sourceforge.net/projects/opencvlibrary/files/opencv-unix/2.4.9/opencv-2.4.9.zipunzip解壓2.安裝依賴sudo apt-get

AWS-G2-Ubuntu16.04安裝cuda,caffe和py-faster-rcnn

G2的顯示卡(GRID K520)驅動:http://www.nvidia.com/Download/Find.aspx 然後參考這個文章安裝CUDA和caffe:http://blog.csdn.net/sinat_26917383/article/details/56

Ubuntu16.04環境安裝CUDAcudnn、Caffe、Tensorflow、pytorch

【主題】Ubuntu16.04環境下安裝CUDA、cudnn、Caffe、Tensorflow、pytorch 【環境】 筆記本: 筆記本:惠普 Pavilion NoteBook 系統環境:Ubuntu16.04+Win10 64位雙系統 顯示卡:NVIDIA  GT

Ubuntu16.04安裝配置Android studiosdkgradle——一篇文章就夠了系列

64位ubuntu16.04完美安裝配置jdk、sdk、Android studio 一、下載安裝jdk、sdk、Android studio 注意:我把JDK安裝在了/opt資料夾下面,/opt資料夾和/usr資料夾同樣都是常用做安裝軟體的資料夾,不同的是/usr是

ubuntu16.04 安裝加入opencv_contribute包的GPU版本的OpenCV3.2(CUDA支援)步驟與問題解決

首先肯定是先安裝依賴了,官方列出了一些: sudo apt-get install build-essential sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev liba

Ubuntu16.04安裝python3.6詳細步驟anaconda安裝詳解

       最近在廖雪峰老師的網站上學習python,開始用的是Windows環境,但是越往後發現用Linux或許更好一些,環境搭建什麼的比較方便,但是對於很少用Linux的同學來說,Linux上python安裝或許會出現各種令人抓狂的問題,本文以Ubuntu16.04為例

Ubuntu16.04安裝python3.6詳細步驟anaconda3安裝詳解

1.Python3.6的安裝  Ubuntu中自帶python2.7和python3.5,但是很多情況下我們需要使用python3.6,以下是在終端中安裝python3.6的詳細步驟: 在終端中輸入下面的命令 wget http://www.python.org/ftp

ubuntu16.04安裝NVIDIA(cuda)-gtx965m相關步驟以及問題

ubunt版本:16.04 筆記本:機械革命,i7-6700,gtx965m(集顯Intel Hm170 ) 安裝NVIDIA以及cuda來測試深度學習 之前安裝ubuntu前對grub檔案進行了修改才進行了安裝,現在進入系統安裝完nvidia後記得將其改

環境配置6-Win10安裝CUDAcuDNN配置caffe

到官網下載相應的安裝包,cuDNN為lib庫,需要先註冊NVIDIA賬號(或者連結複製到迅雷下載)。https://developer.nvidia.com/rdp/cudnn-archivehttps://developer.nvidia.com/cuda-80-ga2-d

ubuntu16.04安裝wineTIM

寫在前面 20180701 想快速使用tim,而不是自己踩雷的朋友,可以在這裡找到可執行程式。 提供在Linux上執行最新版騰訊QQ與TIM的解決方案 Easiest Way to Run QQ&TIM on Linux 體驗還行,執行非常方便,

ubuntu16.04安裝配置nodejsnpm

tps 下載 clas con 官網 down 。。 chm 網上 官網下載ubuntu對應的二進制文件(Linux Binaries (x64)這一行) https://nodejs.org/en/download/ 解壓並安裝 ``` $tar zxvf node

ubuntu16.04 安裝phpMyAdmin

image ubuntu16 log 一個 workbench 過程 cnblogs share enc 由於 ubuntu上沒有什麽好的數據庫界面,當然 說mysql_workbench 的大神不要見笑, 本人英語水平有限,雖然用起來沒有什麽影響,但是就是感覺不舒服 於是

Ubuntu16.04安裝elasticsearch+kibana實現php客戶端的中文分詞

lba 實例 exc common adding creat 啟動服務 uid dbms 1.下載安裝elasticsearch和kibana wget https://artifacts.elastic.co/downloads/elasticsearch/elasti

ubuntu16.04安裝emacs

輸入 mac aps pda posit ins pos snap ubunt 1.直接在軟件中心下載. 2.終端下載: 依次輸入這三條命令即可 sudo add-apt-repository ppa:ubuntu-elisp/ppasudo apt-get updates

Ubuntu16.04安裝redis並實現helloworld

ber -a lang 配置 ges image download rep mon 原文出處:http://blog.csdn.net/xiangwanpeng 1 sudo wget http://download.redis.io/releases/redis-3.2.

Ubuntu16.04安裝Chrome出現“未安裝軟件包 libappindicator1”問題的解決辦法

googl spa 軟件包 依賴 mic 安裝軟件 ren for 其中 1. 強制安裝chrome sudo dpkg -i google-chrome-stable_current_i386.deb --force 2. 補齊依賴 sudo apt-get i