1. 程式人生 > >ubuntu14.04下安裝anaconda3+tensorflow-gpu+升級cuda+升級cudnn

ubuntu14.04下安裝anaconda3+tensorflow-gpu+升級cuda+升級cudnn

一、初始環境介紹

1.主機ubuntu14.04、自帶了python2.7和python3.4,我們在後面也會用到python,但是我們在這裡不用管這個python的。

2.系統以前裝有cuda7.5和cudnn4.x.我們在後面需要升級這兩個東西。

3.系統之前已經裝過顯示卡驅動了。這裡我們不介紹裝顯示卡的過程了,自行百度。

4.上一張我們配置深度學習環境的版本對應圖

根據這張圖的版本對應關係進行相應的安裝。這個GCC版本我經過實踐好像沒什麼影響。

二、 anaconda3的安裝

1.檔案準備:下載anaconda3-4.2.0-linux-x86_64.sh檔案,這個檔案是基於python3.5的,而且下載後GCC是4.4.7的。

網頁連線點選這裡。

2.安裝:下載完成以後在終端利用cd命令進入下載的目錄,然後執行bash Anaconda3-4.2.0-Linux-x86_64.sh ,在安裝的過程我們只需要yes即可,這樣在~/.bashrc就加上了這個路徑。通過一個source ~/.bashrc命令即可啟用。

bash Anaconda3-4.2.0-Linux-x86_64.sh
source ~/.bashrc

3.安裝成功:在終端輸入python,會出現以下圖片。

三、tensorflow-gpu安裝

1.在進行安裝之前可以先進行一下簡單換源:1)更換pip源 2)更換conda源

1)程式碼如下


cd

sudo mkdir .pip

sudo vi ~/.pip/pip.conf

在開啟的檔案中新增如下程式碼


[global]

index-url = https://pypi.douban.com/simple

2)在終端執行sudo gedit ~/.condarc,在開啟的檔案中寫上以下程式碼,儲存即可。

channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
- defaults
show_channel_urls: true

2. conda環境安裝tensorflow-gpu

1)首先在終端輸入程式碼建立一個conda環境:conda create -n tensorflow-gpu python=3.5

這個python的版本號,我們在終端輸入python即可看見。

2)環境建立完成以後,我們通過一下程式碼關閉和開啟環境。

source activate tensorflow-gpu #啟用環境

source deactivate tensorflow-gpu #關閉環境

3)在輸入程式碼 source activate tensorflow-gpu啟用環境後,輸入一下程式碼來安裝tensorflow.

source activate tensorflow-gpu
pip install tensorflow-gpu==1.0.1

這裡我安裝的是1.0.1版本的,也可以安裝1.2.0版本的,主要是按照前面的所發的版本對應關係。

4)等安裝完成以後,在啟用環境的情況下,輸入python,然後輸入import tensorflow as tf,如果沒有報錯的話,那麼說明安裝成功。

import tensorflow as tf

四、cuda升級

  1.因為之前我的電腦上裝了cuda7.5,所以現在需要更新到cuda8.0.這裡我採取的方法主要是刪除7.5,安裝8.0。

1)解除安裝舊的cuda:這裡只需要輸入以下程式碼即可

sudo /usr/local/cuda-7.5/bin/uninstall_cuda_7.5.pl
sudo rm -rf /usr/local/cuda-7.5

2) 下載並安裝cuda8.0:下載網址點這裡   下載以後到下載目錄,在終端輸入以下命令進行安裝。

sudo sh cuda_8.0.61_375.26_linux.run

這個過程需要我們一直按照enter,一直到100%。然後就是1)accept  2) n   3)y 4)y 5)y.然後就是等待安裝完成。

如果是第一次裝這個cuda,那麼在最開始還需要一些驗證命令來驗證是否滿足安裝要求。具體的詳情點這裡

3)設定環境變數

sudo gedit /etc/profile

//在檔案結尾新增

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64/:/usr/local/cuda/lib64:$LD_LIBRARY_PATH

export PATH=/usr/local/cuda-8.0/bin:/usr/local/cuda/bin:$PATH


然後在終端輸入 source /etc/profile 重新啟用一下環境。

4)檢查是否安裝成功,在終端輸入nvcc --version 出現如下圖片即為安裝成功。

五、cudnn升級.

1.同cuda一樣是刪除以前的版本,安裝cudnn5.1.

1)刪除舊版本的cudnn 

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn

2)下載並安裝cudnn:下載點這裡  ,下載完成以後進行解壓,然後用cd命令進入到解壓目錄之下。輸入以下程式碼安裝。

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/

3)建立個軟連線,在終端進入到/usr/local/cuda/lib64 ,然後執行下面的命令,這裡檔案裡的後面數字不對的自行更改。

sudo chmod +r libcudnn.so.7.0.5
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7  
sudo ln -sf libcudnn.so.7 libcudnn.so     
sudo ldconfig

4)驗證安裝:在終端輸入

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

出現版本號即為安裝成功。