【深度学习】CentOS 7 安装GPU版Tensorflow教程(一)

来源:互联网 发布:舌苔厚口臭 知乎 编辑:程序博客网 时间:2024/06/05 02:43

        之前一直在玩cpu版的tensorflow,这些天突然心血来潮,想搞个gpu版的tensorflow来尝尝鲜,没想到把所有能够踩的坑几乎全部踩了一遍,在这里把自己踩的坑和一些安装细节拿出来分享给大家,以便让各位同道中人不要把太多的时间浪费在安装这件看似很easy的事情上来。


安装流程:
1.关闭UEFI
       在BIOS里面检查你的UEFI是否开启,如果开启的话请立马关掉它(这个很湿重要,因为它很有可能导致你的kernel安装失败,笔者就遇到了这个坑,浪费了好多时间),具体怎么关掉就不多说,由于每种电脑型号的BIOS都有所不同。


2.确认自己的显卡支持cuda

[littlebei@localhost ~]$ lspci | grep -i nvidia01:00.0 VGA compatible controller: NVIDIA Corporation GM107 [GeForce GTX 745] (rev a2)01:00.1 Audio device: NVIDIA Corporation Device 0fbc (rev a1)

如果有以上信息出现,说明你的显卡是支持cuda的。


3.确认Linux版本是否支持cuda

[littlebei@localhost ~]$ uname -m && cat /etc/*release

若有信息输出,说明是支持的。


4.检查gcc是否安装

[littlebei@localhost ~]$ gcc --versiongcc (GCC) 4.8.5 20150623 (Red Hat 4.8.5-16)Copyright (C) 2015 Free Software Foundation, Inc.This is free software; see the source for copying conditions.  There is NOwarranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

若出现以上信息说明gcc已经安装。

若没有安装,可以使用一下命令安装

[littlebei@localhost ~]$ sudo yum install gcc gcc-c++


5.安装kernel-devel和kernel-headers

$ sudo yum install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
其中$(uname -r)就是个参数,uname -r表示kernel的版本号


6.关闭X server
$ systemctl stop gdm.service


7.禁用nouveau(因为它是一般linxu系统自带的显卡驱动,会和nvidia冲突,所以必须要关掉)
(1)将 nouveau 驱动加入黑名单:
在 /usr/lib/modprobe.d/dist-blacklist.conf 中加入 blacklist nouveau(这种方式仅限在centos 7,其他Linux 系统自行解决)。
(2)备份 initramfs 文件:

$ sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
(3)重建 initramfs 文件
$ sudo dracut -v /boot/initramfs-$(uname -r).img $(uname -r)



8.关机重启



剩余步骤请查看CentOS 7 安装GPU版Tensorflow教程(二)

















阅读全文
1 0