參考https://blog.csdn.net/u012325865/article/details/73034018
1尿庐、檢查是否安裝了GPU
lspci | grep -i nvidia
看到GPU顯卡型號是GeForce GTX 1080 Ti忠怖,另一張是集顯
2、下載對應(yīng)GPU 顯卡的驅(qū)動
在官網(wǎng)下載對應(yīng)的驅(qū)動http://www.nvidia.cn/Download/index.aspx?lang=cn
要選擇自己GPU對應(yīng)型號的驅(qū)動抄瑟,這里盜了張圖凡泣。
然后跳轉(zhuǎn)到下載頁面,下載 NVIDIA-Linux-x86_64-390.67.run
3皮假、下載CUDA
https://developer.nvidia.com/cuda-toolkit-archive
然后跳轉(zhuǎn)到下載頁面鞋拟,下載 cuda_8.0.61_375.26_linux.run
4、安裝gcc惹资、g++編譯器
root登陸
yum install gcc? ?
yum install gcc-c++
5贺纲、安裝kernel-devel和kernel-headers
yum install kernel-devel?
yum install kernel-headers
6、賦予權(quán)限
chmod 755 NVIDIA-Linux-x86_64-390.67.run
chmod 755 cuda_8.0.61_375.26_linux.run
7褪测、禁用nouveau
系統(tǒng)直接顯示1366X768的分辨率猴誊,就是默認(rèn)裝了驅(qū)動的潦刃,但不是NVIDIA的官方驅(qū)動。Centos默認(rèn)安裝了開源的nouveau kernel driver, 與NVIDIA驅(qū)動沖突懈叹。所以需先退出圖形界面并將nouveau禁用后乖杠,才能安裝NVIDIA官方驅(qū)動。
參考 https://blog.csdn.net/zhangtong11111/article/details/7913197
1)退出圖形界面澄成,進(jìn)入init 3文本模式
Ctrl+Alt+F2? 進(jìn)入純命令模式
登陸----獲取root權(quán)限?
輸入init 3?
2)在/etc/modprobe.d/blacklist.conf中添加兩行內(nèi)容:
blacklist nouveau
optionsnouveau modeset=0
2)給當(dāng)前鏡像做備份
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
3)建立新的鏡像
dracut /boot/initramfs-$(uname -r).img $(uname -r)
4)重啟系統(tǒng)
5)進(jìn)入init 3文本模式滑黔,檢查nouveau driver是否沒有被加載
lsmod | grep nouveau
8、安裝驅(qū)動和CUDA
切換至安裝包文件夾
./NVIDIA-Linux-x86_64-390.67.run? #根據(jù)提示安裝?
./cuda_8.0.61_375.26_linux.run? #根據(jù)提示安裝环揽,Ctrl+C跳過協(xié)議,是否安裝driver這里一定選no,因?yàn)樯弦徊揭呀?jīng)安裝了驅(qū)動了庵佣,我試過選yes后:自動安裝驅(qū)動并提示需要重啟歉胶,重新執(zhí)行CUDA安裝,又會卡在這里巴粪,循環(huán)安裝驅(qū)動-重啟通今。安裝路徑我使用默認(rèn),這個路徑你要記得肛根,后續(xù)配置環(huán)境變量會使用到辫塌。
這里介紹我遇到的兩種錯誤和解決方法。
1)假如安裝驅(qū)動報錯:
ERROR: The Nouveau kernel driver is currently in use by your system. This
? ? driver is incompatible with the NVIDIA driver, and must be disabled
? ? before proceeding. Please consult the NVIDIA driver README and your
? ? Linux distribution's documentation for details on how to correctly
? ? disable the Nouveau kernel driver.
解決方法:禁用Nouveau顯卡驅(qū)動派哲,可以參考我前面的步驟臼氨,也可以參考https://blog.csdn.net/stu741/article/details/40046397
2)假如安裝CUDA報錯:
The driver installation is unable to locate the kernel source. Please make sure that the kernel source packages are installed and set up correctly.
If you know that the kernel source packages are installed and set up correctly, you may pass the location of the kernel source with the '--kernel-source-path' flag.
sudo yum install epel-release
yum install --enablerepo=epel dkms
建議不要嘗試升級內(nèi)核,我試過然后系統(tǒng)崩潰了芭届,一切重來储矩。
9、配置環(huán)境變量
gedit?~/.bashrc
#寫入bashrc文件保存??
#gpu?driver??
export?CUDA_HOME=/usr/local/cuda-8.0??
export?PATH=/usr/local/cuda-8.0/bin:$PATH??
export?LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH??
export?LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib:${LD_LIBRARY_PATH}"?
source之后查看安裝結(jié)果
source?~/.bashrc??
nvidia-smi??
11褂乍、測試cuda
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
./deviceQuery
安裝成功持隧!
12、安裝cudnn7.5
從官網(wǎng)上(https://developer.nvidia.com/cudnn)下載相關(guān)版本的CUDNN(需要先注冊賬號才能下載):一定要選擇CUDA相對應(yīng)版本的逃片。
解壓并拷貝到系統(tǒng)目錄下:
切換至所在目錄
tar xzvf?cudnn-8.0-linux-x64-v6.0.tgz
cp cuda/include/cudnn.h /usr/local/cuda/include
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
13屡拨、查看CUDA,CUDNN 版本號
查看 CUDA 版本:
cat /usr/local/cuda/version.txt
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
14褥实、安裝Anaconda3
1)清華鏡像源下載安裝包:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
2)執(zhí)行安裝呀狼,安裝路徑不要用默認(rèn)的/root/anaconda3,因?yàn)檫@個路徑是root權(quán)限的性锭,安裝后每次使用anaconda3都要切換用戶到root赠潦,非常不方便。因此最好是將anaconda安裝到用戶權(quán)限目錄草冈。
修改/opt目錄及其子目錄的所有者為用戶:chown -R? user1 /opt 目錄opt用戶改成user1,包括dir里面的子目錄和文件她奥。
anaconda安裝路徑改為/opt/anaconda3瓮增,不添加到環(huán)境變量(稍后手動添加),不安裝Microsoft VSCode哩俭。
bash Anaconda3-5.0.1-Linux-x86_64.sh
3)安裝完畢后手動添加環(huán)境變量:
切換回用戶賬戶绷跑,修改用戶的環(huán)境變量 gedit ~/.bashrc,在文件末尾添加一行:export PATH=/root/anaconda3/bin:$PATH凡资,再使其生效source ~/.bashrc砸捏。
修改系統(tǒng)環(huán)境變量 gedit /etc/profile,打開profile文件
在文件末尾添加一行:export PATH=/root/anaconda3/bin:$PATH隙赁,其中垦藏,將“/root/anaconda3/bin”替換為你實(shí)際的安裝路徑,保存伞访。
使環(huán)境變量生效 source /etc/profile掂骏,發(fā)現(xiàn)python改為3.6版本了。
15厚掷、安裝Tensorflow 1.3
pip install tensorflow==1.3