1. 程式人生 > >CentOS7.3安裝NVIDIA-1080ti驅動、cuda、cudnn、TensorFlow

CentOS7.3安裝NVIDIA-1080ti驅動、cuda、cudnn、TensorFlow

可能 點擊 自己 pkg pep drive ron oca detail

  Ubuntu非要換centOS。。。好吧。。。

  看了很多是通過ELRepo源安裝驅動,不過我沒有成功,只是通過它的步驟檢測了顯卡驅動型號。 0、安裝gcc
yum -y install gcc-c++
  此處是重點:如果有之前的NVIDIA驅動請先卸載,而且,要先裝cuda再裝驅動。。。你也可以按照我的步驟來,最後再重裝一次驅動。 1、檢測顯卡驅動及型號   ##添加ELPepo源
$ sudo rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
$ sudo rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-2.el7.elrepo.noarch.rpm
  技術分享   ##安裝NVIDIA驅動檢測
$ sudo yum install nvidia-detect
$ nvidia-detect -v
  兩塊顯卡驅動都是384.59   登錄NVIDIA官網http://www.geforce.cn/drivers設置驅動檢索條件(註意盡量設置語言英文):   技術分享技術分享   檢索結果出現384.59,點擊下載獲取下載鏈接http://us.download.nvidia.com/XFree86/Linux-x86_64/384.59/NVIDIA-Linux-x86_64-384.59.run   ssh到服務器,下載驅動:
$ cd ~/downloads
$ 
wget -r -np -nd http://us.download.nvidia.com/XFree86/Linux-x86_64/384.59/NVIDIA-Linux-x86_64-384.59.run
  可能需要sudo權限,建議通過su root直接到root賬戶下執行安裝。安裝過程中選擇“Accept”,並且會詢問是否要配置啟動參數,選擇“yes”。 2、顯卡沖突   因為NVIDIA驅動會和系統自帶nouveau驅動沖突,執行命令查看該驅動狀態:
$ lsmod | grep nouveau
技術分享   修改/etc/modprobe.d/blacklist.conf 文件,以阻止 nouveau 模塊的加載,如果系統沒有該文件需要新建一個,這裏使用root權限,普通用戶無法再在/etc內生成.conf文件。
$ su
root # echo -e "blacklist nouveau\noptions nouveau modeset=0" > /etc/modprobe.d/blacklist.conf
3、 重新建立initramfs image文件
# mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
# dracut /boot/initramfs-$(uname -r).img $(uname -r)
4、進入NVIDIA目錄執行安裝(建議推遲到cuda安裝後再裝驅動)
$ chmod +x NVIDIA-Linux-x86_64-384.59.run
$ sh NVIDIA-Linux-x86_64-384.59.run
   如果安裝完成,運行命令查看顯卡狀態
$ nvidia-smi
5、安裝cuda   官網下載cuda-rpm包https://developer.nvidia.com/cuda-downloads,一定要對應自己的版本。
$ sudo rpm -i cuda-repo-rhel7-8-0-local-ga2-8.0.61-1.x86_64.rpm
$ sudo yum clean all
$ sudo yum install cuda
  報錯:
Error: Package: 1:nvidia-kmod-375.26-2.el7.x86_64 (cuda-8-0-local-ga2) Requires: dkms You could try using --skip-broken to work around the problem You could try running: rpm -Va --nofiles --nodigest
  缺少2個包,裝第一個:
sudo vim /etc/yum.repos.d/linuxtech.testing.repo
  輸入:
[linuxtech-testing] name=LinuxTECH Testing baseurl=http://pkgrepo.linuxtech.net/el6/testing/ enabled=0 gpgcheck=1 gpgkey=http://pkgrepo.linuxtech.net/el6/release/RPM-GPG-KEY-LinuxTECH.NET
sudo yum --enablerepo=linuxtech-testing install libvdpau
  第二個:
yum -y install epel-release
yum -y install --enablerepo=epel dkms
  OK,再次執行install安裝cuda   。。。發現驅動沒了。。。只好重新安裝,看樣子要先裝cuda再裝NVIDIA驅動,只好再裝一波驅動。 6、測試cuda
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
  技術分享   安裝成功 7、cuda添加到bashprofile中
vim .bashprofile
PATH=$PATH:$HOME/bin:/usr/local/cuda/bin LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64/ CUDA_HOME=/usr/local/cuda export PATH export LD_LIBRARY_PATH export CUDA_HOME
  查看nvcc版本號
nvcc -V
8、安裝cudnn   cudnn官網無法wget,只能在Windows下載然後通過ssh傳過去,為什麽不用U盤或者共享文件夾?因為我在家。。。服務器剛配置。。。不嫌麻煩用xshell插件也可以。這裏安裝了一個Windows下ssh窗口傳輸文件的工具:
yum install lrzsz
  運行命令為rz,不用scp了,很方便。   技術分享   解壓cudnn到cuda的安裝目錄
tar -xvf cudnn-7.5-linux-x64-v5.0-ga.tgz -C /usr/local

9、安裝TensorFlow   如果沒有pip請安裝pip,記得修改一下pip的源,否則速度很慢
yum install python-pip
pip install --upgrade pip
  安裝TensorFlow-gpu版本
pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.0-cp27-none-linux_x86_64.whl 
  測試:   技術分享 成功。over。 參考: http://blog.csdn.net/renhanchi/article/details/73468982 http://www.cnblogs.com/snake553/p/7159376.html http://www.linuxidc.com/Linux/2017-03/141644.htm http://blog.csdn.net/wang2008start/article/details/71319970 此外,建議參考TensorFlow官網,在虛擬環境下安裝。

CentOS7.3安裝NVIDIA-1080ti驅動、cuda、cudnn、TensorFlow