從零開始Ubuntu 16.04 + CUDA 9.0 + CUDNN 7.0 深度學習環境搭建

1. 上手第一件事,更新apt源

sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak

(備份是好習慣)

sudo vim /etc/apt/sources.list

刪光,粘貼以下代碼

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse
deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted universe multiverse
# 預發佈軟件源,不建議啓用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-proposed main restricted universe multiverse

保存並退出

sudo apt-get update
sudo apt-get upgrade

更新一下軟件和庫,然後裝一些常用的東西比如gcc之類的


2. 安裝顯卡驅動
首先要知道自己的顯卡型號,並去NVIDIA官方網站找到對應的驅動,如果不知道的話用

lspci|grep NVIDIA

查看顯卡型號,如果依然看不到什麼型號,去找相應系列的穩定版驅動。

  • 使用第三方源安裝
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update

#添加第三方倉庫並更新源
sudo apt-get remove --purge nvidia*

#卸載老版驅動
sudo apt-cache search nvidia-*

#查看ppa安裝支持的驅動版本,如果有要找的驅動
sudo apt-get install nvidia-#

#直接安裝,其中#是對應的版本號
  • 直接在NVIDIA官網下載對應驅動的 .run 文件,右鍵複製下載鏈接
wget "link-to-driver"

#下載下來之後
sudo chmod 777 driver.run

#改權限
sudo ./driver.run --uninstall

#卸載掉之前的舊驅動
sudo ./driver.run

#直接運行,這裏可能會報兩個錯,一個是kernel的問題,一個是linux頭文件的問題
sudo apt-get install linux-headers-$(uname -r)
sudo apt-get install gcc build-essential

#安裝頭文件和build-essential解決,後面的一路選是就好了。
#如果是桌面版的ubuntu(有圖形界面),在這樣安裝完顯卡驅動之後可能會出現顯示分辨率有問題的bug,是因爲NVIDIA驅動與X config不兼容的問題。
#解決方法:ctrl + alt + f1 進入tty模式,登錄之後
sudo ./driver.run --uninstall

#卸載掉驅動
sudo service lightdm stop

#關閉圖形服務
sudo ./driver.run --no-x-check --no-nouveau-check --no-opengl-files

#在安裝驅動時關掉X config,關掉默認驅動nouveau,不安裝opengl文件,然後一路選是,在變更Xconfig的時候選擇否
sudo service lightdm start

#開啓圖形服務

最後運行

nvidia-smi

如果能看到顯卡狀態,說明安裝成功


3. 安裝CUDA
 

#從官網下載CUDA對應版本和對應操作系統的runfile(local),右鍵複製鏈接地址
wget "link-to-cuda"
#下載下來之後
sudo chmod 777 cuda.run
#更改權限
sudo ./cuda.run
#開始安裝,一路選是,除了第二個問題,問是否要安裝某某版本的顯卡驅動,選否,因爲上面我們已經安裝過了,安裝完成之後
vim ~/.bashrc
#給單個用戶或者
sudo vim /etc/profile
#給所有用戶添加環境變量
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
#保存並退出
source ~/.bashrc
#或者
source /etc/profile
#使環境變量生效
echo $PATH
echo $LD_LIBRARY_PATH
nvcc -V
#檢查是否安裝完成

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章