系統(tǒng):Ubuntu16.04 LTS
顯卡:GeForce GTX 1060
1 安裝NVIDIA顯卡驅(qū)動
1.1 下載NVIDIA顯卡驅(qū)動程序
從(https://www.geforce.cn/drivers)上選擇支持本機顯卡的的驅(qū)動程序下載,這里選擇NVIDIA-Linux-x86_64-384.98.run藻懒,下載至~/Downloads文件夾
1.2 安裝
首先屏蔽nouveau啟動嬉荆,這里選擇blacklist方式鄙早,將nouveau驅(qū)動添加到黑名單blacklist.conf中
#打開blacklist文件
sudo vi/etc/modprobe.d/blacklist.conf
#在文件末尾添加以下語句
blacklist nouveau
#關閉文件后更新
sudo update-initramfs -u
#重啟系統(tǒng)
reboot
#確認nouveau已被禁止
lsmod | grep nouveau
按Ctrl + Alt + F1到控制臺,關閉當前圖形環(huán)境
輸入用戶名和密碼登錄
#以下三句確保結(jié)束x-window服務
sudo init 3
sudo rm -r /tmp/.X*
sudo service lightdm stop
#進入驅(qū)動下載文件夾
cd ~/Downloads
#改文件權(quán)限為可執(zhí)行文件
chmod +x NVIDIA-Linux-x86_64-384.98.run
#執(zhí)行文件扳缕,--no-opengl-files只安裝驅(qū)動文件,不安裝OpenGL文件
sudo ./NVIDIA-Linux-x86_64-384.98.run --no-opengl-files
#根據(jù)提示完成安裝即可,后重啟
reboot
#檢查驅(qū)動是否安裝成功
nvidia-smi
輸出以下信息說明安裝成功
2 配置tensorflow-gpu
2.1 安裝CUDA
2.1.1 下載CUDA
從https://developer.nvidia.com/cuda-80-ga2-download-archive下載CUDA8.0粥庄。CUDA已更新至9.0,未防止出現(xiàn)不兼容性训堆,建議下載舊版坑鱼。網(wǎng)頁如需注冊鲁沥,可注冊一下画恰,注冊到通過可能需要幾分鐘到十幾分鐘缠局。
2.1.2 安裝
#改變文件權(quán)限
chmod u+x cuda_8.0.61_375.26_linux.run
#執(zhí)行文件
sudo ./cuda_8.0.61_375.26_linux
#接下來進入CUDA的安裝過程甩鳄,按q鍵跳過license說明
#accept接受協(xié)議
#是否安裝NVIDIA啟動選擇否(重要!CUDA自帶的顯卡驅(qū)動兼容性差揖赴,而且我們之前已經(jīng)手動安裝了顯卡驅(qū)動)
Istall NVIDIA AcceleratedGraphics Driver for Linux-X86_64 367.48?
(y)es/(n)o/(q)uit: #請選擇n
#其他選項按y或默認即可
2.2 安裝cuDNN
2.2.1 下載cuDNN
從(https://developer.nvidia.com/rdp/cudnn-download)下載cuDNN5.1到~/Downloads文件夾燥滑。同樣為防止出現(xiàn)不兼容性,建議下載舊版恃锉,網(wǎng)頁如需注冊肪跋,可注冊一下州既。
2.2.2 安裝
#進入剛剛cuda的安裝目錄
cd /usr/local
#解壓后自動配置
sudo tar -zxvf ~/Downloads/cudnn-8.0-linux-x64-v5.1.tgz
#設置環(huán)境路徑
vi ~/.bashrc
#在文件末尾添加以下三個語句
exportLD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
exportCUDA_HOME=/usr/local/cuda-8.0
exportPATH=/usr/local/cuda-8.0/bin:$PATH
#關閉文件后更新
source ~/.bashrc
#確認完成CUDA安裝
nvcc -V
輸出以下信息說明安裝成功
2.3 安裝tensorflow-gpu
系統(tǒng):Ubuntu16.04LTS
python:python2.7
tensorflow官網(wǎng)上給出了virtualenv、“native” pip晤郑、Docker磕洪、Anaconda和源代碼安裝五種方式析显,這里推薦使用virtualenv安裝
#安裝pip和virtualenv
sudo apt-get install python-pip python-dev python-virtualenv
#創(chuàng)建virtualenv環(huán)境分尸,命令中“~/tensorflow”可更改為自定義路徑
virtualenv --system-site-packages ~/tensorflow
#激活virtualenv環(huán)境孔庭,之后每次使用tensorflow之前圆到,都需要首先激活環(huán)境
source ~/tensorflow/bin/activate
#命令行改變?yōu)椤?tensorflow)$”
#確保pip版本在8.1以上
easy_install -U pip
#安裝tensorflow-gpu芽淡,tensorflow默認安裝1.4版本,未避免出現(xiàn)不兼容性白胀,這里推薦安裝1.2的較低版本
pip install --upgrade tensorflow-gpu==1.2
#完成安裝纹笼,運行以下程序確認tensorflow-gpu可以運行
python
import tensorflow as tf
hello = tf.constant('Hello,TensorFlow!')
sess = tf.Session()
出現(xiàn)如下信息廷痘,說明tensorflow-gpu啟動成功
#輸出“Hello, TensorFlow!”說明tensorflow成功運行
print(sess.run(hello))
tensorflow的安裝說明主要來自tensorflow官網(wǎng),如果安裝過程出現(xiàn)問題兄猩,請參閱(https://www.tensorflow.org/install/install_linux)以獲得更多信息和幫助。
鑒于筆者水平有限铜秆,以上安裝步驟如有錯誤或疏漏之處核蘸,煩請不吝指教