centos7安裝tensorflow_gpu完整教程

配置信息

麗臺K2200+顯卡驅動440.31+cuda9.0+cudnn7.6.2+python3.6+tensorflow_gpu1.12.0(試試這個)

 

1、安裝顯卡驅動

1.1查看顯卡型號,命令:lspci  | grep -i vga,

1.2根據顯卡型號下載顯卡驅動,下載地址:https://www.nvidia.cn/Download/index.aspx?lang=cn

1.3安裝顯卡驅動

(1)安裝依賴環境:yum install gcc kernel-delve -y(要確保內核版本號和內核源碼版本號一致,查看內核版本:ls /boot | grep vmlinu,查看源碼包版本:rpm -aq | grep kernel-devel)

(2)屏蔽系統默認的mouveau,root用戶打開gedit /lib/modprobe.d/dist-blacklist.conf,註釋掉#blacklist nvidiafb,添加blacklist nouveau和options nouveau modeset=0(查看mouveau是否禁用:lsmod | grep nouveau)

(3)重建initramfs image,兩條命令如下:

mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

dracut /boot/initramfs-$(uname -r).img $(uname -r)

(4)修改運行級別爲文本模式並重啓:systemctl set-default multi-user.target和reboot

(5)進入下載的驅動所在目錄,修改執行權限:chmod +x NVIDIA-Linux-x86_64-346.47.run,執行安裝:./NVIDIA-Linux-x86_64-346.47.run

(6)回到圖像模式並重啓systemctl set-default graphical.target和reboot

(7)驗證nvidia-smi

2、安裝cuda

2.1下載cuda,下載地址:https://developer.nvidia.com/cuda-downloads(cuda10),https://developer.nvidia.com/cuda-90-download-archive(cuda9)可以用wget也可以先下載下來)

2.2修改運行級別爲文本模式並重啓:systemctl set-default multi-user.target和reboot

2.3 進入下載的驅動所在目錄,修改執行權限chmod +x cuda_8.0.44_linux.run,執行:./cuda_8.0.44_linux.run

2.4安裝中的設置

 

2.5回到圖像模式並重啓systemctl set-default graphical.target和reboot

2.6設置環境變量:在/etc/profile文件中添加三行:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

export CUDA_HOME=/usr/local/cuda

,並執行source /etc/profile使其立即生效。(注意這裏的路徑就是cuda,而不是cuda-9.0)

2.6驗證 nvcc -V

3、安裝cudnn

3.1下載cudnn,下載地址:https://developer.nvidia.com/rdp/cudnn-archive

3.2解壓tar -xvf cudnn-8.0-linux-x64-v6.0.tgz -C /usr/local/  (貌似只解壓,不知道徐不需要複製,複製參考https://www.jianshu.com/p/a201b91b3d96

4、安裝python3.6

4.1解決基礎依賴yum install -y openssl-devel openssl-static zlib-devel lzma tk-devel xz-devel bzip2-devel ncurses-devel gdbm-devel readline-devel sqlite-devel gcc libffi-devel

4.2下載python:wget https://www.python.org/ftp/python/3.6.0/Python-3.6.0.tgz

4.2解壓並遷移tar -xvf Python-3.6.0.tgz 和mv Python-3.7.0 /usr/local

4.3執行配置文件並編譯cd Python-3.6.0和 ./configure和make&&make install

4.4替換之前版本的python

建立軟連接mv /usr/bin/python /usr/bin/python_bak和ln -s /usr/local/Python-3.6.0/python /usr/bin/python,輸入python看是不是python3.6.0(替換之前版本後,因爲系統默認是python2.x,所以需要把一些文件中python 改爲python2.7)

4.5、安裝pip並修改源(可以不做,直接在pip3命令上加參數即可):

yum install python-pip和pip install --upgrade pip(試一下pip和pip3兩個命令,貌似也都需要修改源)

[global]

index-url = http://mirrors.aliyun.com/pypi/simple/

[install]

trusted-host=mirrors.aliyun.com

 

5、pip安裝tesnsorflow

pip3 install tensorflow-gpu==1.12.0

不配置pip源可以使用pip3 install tensorflow-gpu==1.12.0 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

 

參考:

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章