【解决】Centos7 安装NVDIA GPU+TensorFlow-gpu1.5.0

最新推荐文章于 2024-04-15 18:30:35 发布

原创

最新推荐文章于 2024-04-15 18:30:35 发布 · 2.1k 阅读

4 ·

CC 4.0 BY-SA版权

由于项目需要，需要安装tensorflow-gpu1.5.0到Centos环境下。

环境介绍：
Centos7
gcc-4.8.5
python3.5

我用的GPU的种类是：GeForce GTX1080 Ti，是英伟达的GeForce系列10的产品。

安装TensorFlow-GPU版本需要先安装后GPU。之前的同事已经将gpu安装完毕，但gpu的驱动和cuda版本都不符合tensorflow-gpu的要求，所以要先卸载掉gpu环境。同事安装的环境是：cuda driver 387，cuda8.0，cudnn6.0

注：TensorFlow-gpu需要cuda9.0，cudnn-7.0版本（本人测试7.1版本不行），并且安装最新的cuda驱动（至于为什么要装最新的，我也不知道）。

背景介绍差不多了，下面开始安装。

1.卸载环境

[root@hadoop1 ~]# pip show tensorflow-gpu
Name: tensorflow-gpu
Version: 1.4.0
Summary: TensorFlow helps the tensors flow
Home-page: https://www.tensorflow.org/
Author: Google Inc.
Author-email: opensource@google.com
License: Apache 2.0
Location: /usr/local/lib/python3.5/site-packages