在Ubuntu搭建TensorFlow环境

Posted 2022-07-31 chszs

在Ubuntu搭建TensorFlow环境

• 版权声明：本文为博主chszs的原创文章，未经博主允许不得转载。

我一直想要最大化TensorFlow的计算能力，于是我决定在家用电脑上安装和优化它。这里，我将分享如何选择安装TensorFlow的某个版本——它并不像看起来那么容易，因为你需要了解系统功能，而且涉及到GPU、以及你正在使用的Python版本。在选择TensorFlow的版本时，可以选择“Release”版本，也可以选择每日构建版。本文讲述如何轻松获得每日构建版并安装到自己的机器上运行。（系统环境是Ubuntu Linux）

在安装TensorFlow前，有四个主要选项：

• Python 2.x
• Python 2.x + GPU
• Python 3.x
• Python 3.x + GPU

带GPU支持的版本可以充分利用GPU的计算能力，但首先要检查您的系统是否支持CUDA GPU。

打开终端，执行命令：

# lspci | grep -i nvidia

看看返回了什么，比如我的：

查看GPU型号名称：GeForce GTX 860M，然后可以在NVIDIA官网检查是否兼容CUDA的信息，具体见：https://developer.nvidia.com/cuda-gpus

可以看到支持我的GPU，所以可以安装带GPU的TensorFlow版本。查看TensorFlow网站，具体见：https://www.tensorflow.org/install/install_linux

可以看到有一个我需要安装的依赖库libcupti。安装它：

# sudo apt-get install libcupti-dev

关于如何安装TensorFlow本身，可以阅读：https://www.tensorflow.org/install/install_linux#InstallingNativePip

使用Nightly Builds版安装TensorFlow

接下来需要确定本机的Python版本以及其相关的pip工具。如果还没有Python环境，应该先安装它们。可以参阅系统说明：https://www.python.org/

Ubuntu Linux默认自带了Python 2.7，就将就这个版本吧，先安装pip工具：

# sudo apt-get install python-pip python-dev

请注意，如果您使用Python 3.x，就应该使用软件包python3-pip和python3-dev。

还要注意，pip工具的版本至少应为8.1以上的版本。（用pip -V检查版本）

一旦完成了这一切，就可以简单地安装Nightly Build版的TensorFlow：

# sudo pip install tf-nightly-gpu

或者，如果您的系统不支持CUDA（请参阅上面的部分），请使用：

# sudo pip install tf-nightly

现在可以使用以下步骤检查TensorFlow是否工作。

首先：打开一个Python解释器：

# python

然后，导入TensorFlow库：

import tensorflow as tf

请注意，如果没有正确安装TensorFlow，或者将带GPU的版本安装在不受支持的系统上，就会在此处出现错误。CUDA错误在这一步上非常普遍。如果有效，就可以试试打印出TensorFlow的版本：

print(tf.__version__)

（注意version前后有2个下划线）

完成后，您应该看到打印出的TensorFlow版本 - 如下所示：

可以看到，由于使用了最新的每日构建版，所以TensorFlow当前的最新版本是1.4。