如何在Windows和Linux上下载、安装和设置Tensorflow

TensorFlow是由谷歌开发的用于机器学习和人工智能的开源平台。它帮助开发人员在这个领域进行各种任务。

首先,您需要了解深度学习或者具体说是深度学习才能使用TensorFlow。

在这里,让我简要介绍一下TensorFlow的一些特点以及在Windows和Linux上快速安装它的方法。

TensorFlow概述

从技术上讲,TensorFlow是一个帮助深度学习应用和其他机器学习用例的平台。

它使构建和部署基于机器学习的应用程序变得简单。如果您想使用机器学习解决一个问题,您可以借助TensorFlow获得帮助。

TensorFlow提供了使用Python或JavaScript开发和训练模型的工具。虽然我不是一个开发人员,但您可以探索它的文档,了解它如何影响您部署机器学习应用程序的工作流程。

TensorFlow的特点

TensorFlow因其多个原因而著名,您可以根据其最佳特性来评估它。

如果我们讨论技术上的好处,您将不得不根据您的实际情况来进行比较。因此,我们将专注于对大多数人有益的常见功能。

1. 开源

谷歌于2015年决定将TensorFlow开源,以便社区进一步改进并提供其工作原理的透明性。

开发人员可以以各种方式自定义库以解决您可能没有预料到的问题。

如果没有开源框架,它可能不会像现在这样受欢迎。因此

2. 简单调试

TensorFlow旨在帮助您轻松构建模型,因此简化的调试体验是其中的一部分。

直观的用户体验是锦上添花。

3. 支持CPU和GPU

使用TensorFlow,您可以在CPU或GPU上训练数据计算。通常,与CPU相比,GPU在深度学习应用程序中速度更快。

因此,如果您的工具库中有一块强大的GPU,TensorFlow可以帮助您充分利用它。

4. 有用的机器学习API

API能够帮助开发人员将各种功能集成到他们的应用程序中。TensorFlow提供了访问一系列稳定API的接口。

其中一些API可能还提供了性能优势。根据其官方声明,您在Python中使用这些API应该没有问题。如果您使用其他语言,您需要与TensorFlow维护人员确认它们是否适用于您的用例。

5. 用于生产的现成模型

TensorFlow提供了各种预训练模型。无论您是专业人士还是新手,您都可以使用这些模型来节省时间,更快地构建机器学习模型。

除了这些特性,您还可以获得灵活性、易用性、可视化工具包等等,这些都可以帮助您进行机器学习开发。

现在您已经对TensorFlow有了很好的了解,那么您可以从哪里下载它?如何在Windows和Linux系统上安装和设置它?

我们来讨论一下下面的内容。

下载和安装TensorFlow

与其他程序不同,您在这里不会得到一个.exe安装文件。主要是您需要使用推荐的软件包管理器下载软件包。

总的来说,有不同的安装方式。我们可以列举如下:

  • 使用Miniconda和pip
  • 在WSL 2上使用Miniconda和pip
  • 使用一个软件包管理器
  • 从源代码构建

如何在Windows上安装TensorFlow?

与其他程序不同,您在这里不会得到一个.exe安装文件。您需要使用推荐的软件包管理器下载软件包。

#1. 使用Miniconda和pip(推荐方法)

注意:在撰写本文时,TensorFlow 2.10是在Windows上支持GPU的最后一个版本(原生支持)。如果您使用更新的软件包,TensorFlow建议您在WSL 2中安装TensorFlow下面将讨论该方法。

如果您想要使用支持GPU的TensorFlow,TensorFlow建议使用Miniconda(conda软件包管理器的安装程序)来开始。

使用Miniconda,您可以创建一个单独的环境,以避免与系统中的任何其他软件发生冲突。

开始之前,您需要下载最新的Miniconda Windows Installer并按照屏幕上的说明完成安装。

完成后,您需要启动Miniconda提示符,如屏幕截图所示:

这就是它的样子:

在看到Anaconda提示窗口后,您可能想要输入以下命令,以确保conda软件包管理器已经更新:

conda update -n base -c defaults conda

完成这些步骤后,您需要按照以下步骤安装TensorFlow:

首先,创建一个新的环境(名称为tf):

conda create --name tf python=3.9 

提示:您可以使用以下命令激活/停用它:conda activate tfconda deactivate

您需要激活它以继续操作。要在过程中启用GPU支持,您必须确保已安装图形驱动程序(NVIDIA GPU),然后使用以下命令安装一些软件包:

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

它会下载大约1 GB的软件包,其中包括用于部署带有GPU和深度神经网络的机器学习应用程序的工具。

最后,您将需要利用pip package manager来安装TensorFlow软件包。您可以选择使用conda来安装Tensorflow,但它可能没有所需的最新稳定版本。

在继续之前,确保使用以下命令更新pip:

pip install --upgrade pip

完成后,使用以下命令安装TensorFlow:

pip install tensorflow

您会注意到有许多软件包正在构建/安装。它可能在过程中显示为停滞不前,但请稍等片刻,它应该会恢复并完成安装。

#2. 在WSL 2上使用Conda和pip

假设您已经在系统上设置了WSL 2,您可以在发行版的终端中使用以下命令安装TensorFlow:

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/

python3 -m pip install tensorflow

# 验证安装:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

您可以一次粘贴所有内容,它将逐个处理。

如果您尚未在Windows上安装WSL 2,请前往以管理员权限运行的命令提示符,然后输入以下内容:

wsl.exe --install

它将下载Ubuntu并为您的系统启用WSL功能。您需要重新启动计算机才能完成安装。

如果在系统上找不到Ubuntu,可以转到Microsoft Store并安装Ubuntu WSL。

#3. 从源代码构建

考虑到TensorFlow是开源的,您可以根据自己的配置选项从头开始构建它。

因此,这只推荐给了解所有选项并了解配置细节的高级用户。请参考official documentation了解更多信息。

如何在Linux上安装TensorFlow?

与Windows一样,您可以使用Miniconda和pip在Linux上安装TensorFlow。或选择从源代码构建。

让我向您展示如何完成:

#1. 使用Miniconda和pip(推荐方法)

注意:按照与Windows相同的命令操作。唯一的区别是在Linux上安装/下载Miniconda的方式。

以下是如何在Linux上使用终端安装Miniconda:

curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o Miniconda3-latest-Linux-x86_64.sh

bash Miniconda3-latest-Linux-x86_64.sh

重新启动Linux发行版上的终端,您将看到如下内容:

您会注意到终端提示变量之前有一个(base)。这表示conda当前处于活动状态并已安装。

除非您完成了TensorFlow的安装,否则不要停用它。

您可以导航到上述针对Windows的步骤并进行安装。或者,粘贴以下内容以安装TensorFlow:

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/

python3 -m pip install tensorflow

# 验证安装:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

在Linux上,您可能会遇到GPU驱动程序的问题。为此,您应该参考NVIDIA’s documentation以获取更多信息。

#2. 从源代码构建

与Windows一样,在Linux上从源代码构建是棘手的,纯粹是为高级用户准备的。

您(假设您是初学者)不应选择此方法,除非您有特定目标。探索更多相关信息的最佳方法是参考documentation

如何使用Docker安装TensorFlow?(Windows和Linux)

无论平台如何,Docker都可以让您轻松安装TensorFlow映像。

确保您的系统上已安装Docker,或者您可以按照我们的Docker installation guide进行帮助。

设置完成后,您需要从Docker中输入以下命令:

docker pull tensorflow/tensorflow  

您需要熟悉Docker容器,以便为您的工作启动具有所需配置的容器。

有关特定的GPU支持或下载不同版本的TensorFlow,请参考official documentation中提供的选项。

当您希望使用Docker运行它时,命令看起来像这样:

docker run [-it] [--rm] [-p hostPort:containerPort] tensorflow/tensorflow[:tag] [command]

TensorFlow常见安装错误

最后,这里有一些在安装TensorFlow时可能遇到的常见错误:

  • ImportError: DLL load failed: The specified module could not be found.“:

这只是意味着CUDA未安装。

  • “AttributeError: module ‘tensorflow’ has no attribute’ Session’”

您的TensorFlow安装已损坏。完全卸载TensorFlow和TensorFlow-GPU,然后重新安装TensorFlow-GPU。

  • 一般问题,可能与tensorboard有关

Tensorflow使用tensorboard,它有时可能是“任性”的。必须确保tensorboard与TensorFlow版本相同,并卸载较新的版本。

  • “ImportError: DLL load failed: The specified module could not be found“:

要么CUDA未安装(使用memcheck),要么安装的TensorFlow GPU版本不兼容。解决方案很简单:卸载TensorFlow-GPU并安装较旧的版本。

  • “ImportError:找不到‘ cudart 64_10.dll’”

安装的CUDA版本与您的计算机不兼容。错误告诉您需要64位的CUDA 10.0。因此,您可以完全卸载CUDA,然后安装控制台中指定的版本。请记住更改环境变量以适应新的安装

结论

TensorFlow的安装只需一次,并且通过我们的指南,对于大多数人来说应该是一个无障碍的过程。

如果您已经有旧的Python版本或旧的Conda软件包管理器的配置或设置,请确保应用最新更新以无缝安装TensorFlow。

您还可以探索最佳的AI Platforms来构建AI和ML应用程序。

类似文章