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安装中的设置

centos7安装tensorflow_gpu完整教程_第1张图片

 

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

centos7安装tensorflow_gpu完整教程_第2张图片

 

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

 

参考:

centos7安装tensorflow_gpu完整教程_第3张图片

centos7安装tensorflow_gpu完整教程_第4张图片

 

你可能感兴趣的:(云计算)