凡對本文有任何疑惑可加QQ群交流:1081332609
一、版本信息
操作系統(tǒng):centos7.x
顯卡型號:GeForce GTX 750 Ti
python版本:python3.6
tensorflow版本:tensorflow-gpu==1.14.0
NVIDIA驅(qū)動:NVIDIA-Linux-x86_64-440.36.run
CUDA:cuda_10.0.130_410.48_linux.run
cudnn: cudnn-10.0-linux-x64-v7.6.4.38.tgz
二歹茶、確認自己的顯卡支持cuda
會顯示顯卡的型號
lspci | grep -i nvidia
三橱野、檢查gcc是否安裝
gcc --version
四、安裝kernel-devel和kernel-headers
其中$(uname -r)就是個參數(shù)幼驶,uname -r表示kernel的版本號床绪。
sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
五婆廊、關(guān)閉X server
systemctl stop gdm.service
六饼疙、停用原有顯卡
打開如下文件
vi /usr/lib/modprobe.d/dist-blacklist.conf
編輯溺森,并在最后添加如下兩行,并保存
blacklist nouveau
options nouveau modeset=0
備份 initramfs 文件
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
重建 initramfs 文件
dracut /boot/initramfs-$(uname -r).img $(uname -r)
重啟電腦
reboot
七、安裝NVIDIA驅(qū)動
1、到官網(wǎng)下載與顯卡及操作系統(tǒng)匹配的顯卡驅(qū)動
官網(wǎng)地址:http://www.nvidia.cn/Download/index.aspx?lang=cn
sudo sh NVIDIA-Linux-x86_64-440.36.run --kernel-source-path=/usr/src/kernels/$(uname -r)
2屏积、安裝界面
在accept的頁面選擇Accept医窿,
在32-bit頁面選擇No
在X- configuration頁面選擇Yes
八、安裝CUDA
1炊林、下載
官網(wǎng)地址:https://developer.nvidia.com/cuda-10.0-download-archive
2姥卢、安裝
sudo sh cuda_10.0.130_410.48_linux.run
3、配置cuda環(huán)境變量
編輯bashrc文件
sudo vi ~/.bashrc
在最后添加如下內(nèi)容铛铁,并保存(注意cuda的版本號)
export CUDA_HOME=/usr/local/cuda-10.0
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib:${LD_LIBRARY_PATH
九隔显、安裝cuDNN
1却妨、下載
下載前需要先注冊登錄饵逐,填寫問卷調(diào)查
官網(wǎng)地址:https://developer.nvidia.com/rdp/cudnn-archive
選擇下載與cuda匹配的cuDNN(注意對應(yīng)好cuda的版本號)
2、安裝
tar -xvzf cudnn-10.0-linux-x64-v7.6.4.38.tgz
cp cuda/include/* /usr/local/cuda/include
cp cuda/lib64/* /usr/local/cuda/lib64
十彪标、安裝tensorflow-gpu
建議采用國內(nèi)鏡像加速安裝
pip install -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com tensorflow-gpu==1.14.0
八年抗戰(zhàn)倍权,終于取得最后勝利!@萄獭薄声!