1 搭建準(zhǔn)備
進(jìn)行深度學(xué)習(xí)主機(jī)的環(huán)境搭建崭庸,首先當(dāng)然是要了解自己的機(jī)器啦慧瘤,下面我們來看看本次搭建的軟硬件配置睬澡。
硬件配置:
機(jī)箱主板:聯(lián)想P910
存儲(chǔ):512G(SSD)+4T(HDD)+32G(RAM)
CPU:2*E5-2630V4
顯卡:TITAN XP*1
軟件配置:
操作系統(tǒng):64位windows7專業(yè)版
深度學(xué)習(xí)開發(fā)比較方便是在linux系統(tǒng)啦曙砂,目前ubuntu16.04是比較穩(wěn)定的版本,支持大多數(shù)的庫狐蜕,親測不會(huì)出現(xiàn)太多的問題宠纯。由于配置了顯卡和后期會(huì)使用CNN,還需要安裝cuda和cuDNN馏鹤,考慮到各版本的相互的兼容性問題征椒,選取比較穩(wěn)定的軟件版本娇哆,如下所示:
ubuntu 16.04
CUDA 9.0?
cuDNN 7.1.1
(各軟件及版本下載鏈接之后給出)
2 開始搭建
2.1 安裝linux單系統(tǒng)(ubuntu16.04)
2.1.1 首先去網(wǎng)易開源鏡像下載ubuntu16.04
2.1.2 下載軟碟通湃累,使用軟碟通和ubuntu的鏡像文件,制作U盤啟動(dòng)盤
具體操作如下:
打開-選擇下好的系統(tǒng)iso文件-選擇u盤-啟動(dòng)-寫入硬盤映像
“寫入方式”選擇“USB-HDD+”
接著單擊“格式化”碍讨,按默認(rèn)即可(一般默認(rèn)文件系統(tǒng)位FAT32)
注意鏡像文件大小治力,應(yīng)小于U盤大小
2.1.3 重啟電腦,在黑屏狀態(tài)下勃黍,按住F12宵统,進(jìn)入U(xiǎn)盤啟動(dòng),進(jìn)入ubuntu安裝引導(dǎo)
ubuntu安裝參見鏈接
安裝時(shí)最好保持機(jī)器在斷網(wǎng)的狀態(tài)下覆获,提高安裝速度
2.2 安裝顯卡驅(qū)動(dòng)+CUDA+cuDNN
2.2.1 下載cuda9.0及安裝
首先马澈,根據(jù)硬件型號(hào)及系統(tǒng)環(huán)境,下載所需資源:(安裝cuda時(shí)會(huì)安裝顯卡驅(qū)動(dòng))
CUDA下載鏈接+安裝方法
https://developer.nvidia.com/cuda-downloads
設(shè)置環(huán)境變量:
sudo gedit /etc/profile
$ export PATH=/usr/local/cuda-7.5/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
2.2.1下載cuDNN及安裝
cuDNN下載鏈接
https://developer.nvidia.com/rdp/cudnn-download
如果是奇怪的后綴弄息,就cp打開
1sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
2sudo cp cuda/include/cudnn.h /usr/local/cuda/include
3sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
4sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
安裝tensorflow-GPU版本
pip install tensorflow-gpu
相關(guān)教程鏈接:
https://www.r-bloggers.com/lang/chinese/2042
Ubuntu16.04 + Titan XP + cuda8.0 + cudnn5.1 + opencv3.3.0 + caffe
http://blog.csdn.net/mynotwo/article/details/78661664
Ubuntu16.04安裝NVIDIA Titan Xp顯卡驅(qū)動(dòng)
http://blog.csdn.net/javahaoshuang3394/article/details/76425009
http://www.52nlp.cn/從零開始搭建深度學(xué)習(xí)服務(wù)器環(huán)境配置ubuntu-1080ti-cuda-cudnn
安裝完cuda要進(jìn)行測試
2痊班、安裝cuda 9.0 deb文件(自帶顯卡驅(qū)動(dòng))
2.1 獲取CUDA安裝包,在NVidia官網(wǎng)下載local deb文件摹量。
2.2執(zhí)行如下命令安裝:
$ sudo dpkg -i cuda-repo-ubuntu1410-7-0-local_7.0-28_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda
2.3設(shè)置環(huán)境變量:sudo gedit /etc/profile
$ export PATH=/usr/local/cuda-9.0/bin:$PATH
$ export LD_LIBRARY_PATH=/usr/local/cuda-7.5/lib64:$LD_LIBRARY_PATH
2.4重啟機(jī)器:用其中一個(gè)樣例進(jìn)行驗(yàn)證測試
$ cuda-install-samples-7.5.sh ~
$ cd ~/NVIDIA_CUDA-Samples_7.5/5_Simulations/nbody
$ make
$ ./nbody
或者在sample目錄中make測試項(xiàng)目1_utility涤伐,再運(yùn)行./deviceQuery
一些小技巧:
1.查詢NVDIA顯卡編號(hào)
lspci | grep -i nvidia
2.顯卡信息
nvidia-smi
3.顯存信息
nvidia-smi -l
3.查詢cuda版本+移除cuda+查詢cuda鏈接目錄
apt search cuda
apt remove cuda
ll /usr/local/cuda
ln 可以用來指向cuda目錄
如果是升級(jí)cuda和cuDNN馒胆,直接覆蓋即可,pip會(huì)自動(dòng)進(jìn)行管理凝果。
python相關(guān)安裝:
sudo apt-get install --no-install-recommends
git graphviz python-dev python-flask python-flaskext.wtf?
python-gevent python-h5py python-numpy python-pil?
python-pip python-protobuf python-scipy