在Win10上安装支持GPU训练的Tensorflow

简介: 本文作者通过他的实际经历手把手教授如何在win10操作系统上安装支持GPU训练的tensorflow。

前面的文章中,我展示了如何测试你的Linux系统,看看你是否有支持TensorFlow的GPU,并在当时承诺接下来介绍Windows和MacOS的相关知识。因此,在本文中,我讲介绍如何测试支持GPU的Windows系统,如果你有这方面的需求,可以在获得TensorFlow的构建并确保其正常工作,本文会介绍安装和配置所需的驱动程序。

a222c71e0a5fd28deb81345a8dde21f2db6514cf

但在你进一步研究学习之前,你还需要了解一下Python pip,而这些通常不会在Windows的机器上出现。

1.获取Python和Pip

有很多方法可以安装Windows版的Python和pip,但是我发现最简单的方法是到Python站点下载可执行文件安装程序

一旦你下载了它,启动它,并确保选择安装的自定义选项。这会将你带入一个配置对话框,你可以选择pip该对话框作为安装选项。

44f0e34fc004036e4c32fdeee647f8b0fc331edc

我建议你只选择他们。

完成安装后,你可以打开命令提示符并键入python以查看你正在使用的版本,从上面我可以看到,我下载了3.6.4,并且我的Python命令与此匹配。

af2b0efdf834bd058a99f9559ea5bc13ee90b594

你可以退出Python解释器:exit()。

然后测试你pip install的:pip -V。

你应该看到这样的东西:

9f6c08dbb1fe7af245a0df66970089a61bb2377d

你现在可以适用你windows对应版本的Python和pip安装程序。你将使用它们来安装TensorFlow,但首先,我们需要检查你的GPU是否支持。

2.检查你的Windows GPU版本:

在命令指示符下,你可以使用以下命令启动设备管理器:

control /name Microsoft.DeviceManager

然后查找“显示适配器”设置,将其打开,然后阅读适配器的名称。你应该看到这样的东西:

8f284d861919e4925278627897b411b6a0d66859

正如你所看到的,我的系统有一个GTX980 Ti。你需要查看你的,并了解下支持CUDA的GPU NVIDIA,网站在这里如果你的GPU正好是GTX980 Ti,那么你知道了TensorFlow可以在你的机器上支持GPU的操作。但是在安装和运行TensorFlow之前,你需要为你的机器安装CUDA驱动,CUDNN更新它。

3.安装CUDA驱动程序:

在写这篇文章的时候,TensorFlow已经建立对CUDA9.0的支持,并且发布版本支持8.0。如果你在此处访CUDA下载网站,你会注意到它会将你带到最新版本(现在是9.1),因此请确保通过选择下面的Legacy Resources链接来下载正确版本的驱动程序在下载界面

f190df8e32cdeb75f164238795d9c06bd6f94be9

当然,随着CUDA和TensorFlow的安装,它也会提醒你(本文稍后介绍),你将不得不返还本网站以获取正确版本。下载系统的驱动程序并使用默认值进行安装。这需要一点点时间,还有几次重启。

在运行TensorFlow之前,你还需要一个与你的CUDA版本相匹配的CuDNN版本,但是我个人觉得它更容易不安装,因为首先你必须确保你拥有正确版本的CUDA,其次你应该先安装TensorFlow并成功让它运行。在安装CuDNN的过程中,有两件事情可能会发生:

1. 它将在CUDA驱动程序上失败,因此你需要重复此步骤。

2. 它会在CuDNN驱动程序上失败,所以你知道你的CUDA驱动程序是好的,你需要得到一个正确的CuDNN版本。

接下来我们介绍如何安装TensorFlow。

4.安装TensorFlow Nightly:

我假设你在使用TensorFlow和GPU,所以要从命令提示符下安装它,只需要输入:

sudo pip install tf-nightly-gpu

然后,一旦完成,在命令提示符窗口中输入:

python

...打开Python编辑器,在其中输入:

import tensorflow as tf

如果你使用的是GPU版本,则应该会出现错误。

如果你有错误的CUPA驱动程序,它可能会显示cudart64_XX.dll失败,其中XX是版本号。在撰写本文时,它cudart_90.dll告诉我,我需要CUDA驱动程序的9.0版本。仔细检查此版本,并重复上一步(下载并安装驱动程序),直到此错误消失。一定要关闭命令提示符窗口,并打开一个新的测试安装,以便你的路径是最新的。

当你有正确的CUDA驱动程序,但错误或缺少CuDNN驱动程序,你会看到一个错误,说cudnn64-X.dll缺少什么东西,在哪里X是一个版本号。记下这一点,你会在下一步中使用它。在继续之前,退出Python并关闭命令提示符窗口。就我而言,它告诉我cudnn64_7.dll缺少了,所以我知道我需要版本7.dll.

5.安装CuDNN库:

CuDNN库是针对深度神经网络的CUDA更新,是tensoeFlow用于加速NVidia GPU上的深度学习库,你可以从这里下载。你必须首先注册一个NVidia开发者账号,这很快就会完成,并且这是免费的。登陆后,你会看到各种CuDNN下载。以下是你必须与上一节中下载的CUDA版本相匹配的位置。所以,我使用了CUDA9.0,

559b5de177049a9c4d40333c7385ff2dc6d2a3a2

所以我确定我使用了一个CuDNN来匹配这个和你在最后一步看到的版本(在我的例子中是版本7)—我为CUDA9.0选择了CuDNN v7.0.5。

6.测试TensorFlow:

如果一切顺利,现在可以打开命令提示符窗口,然后输入“python”打开Python解释器。然后输入:

import tensorflow as tf

如果你有正确版本的,你应该看到没有错误,像这样

6c28906134da7a532a0209451e3ca153a4d77dbe

如果你看到错误cudart64_xx.dllcudnn64_x.dll,回到前面的步骤,并确保你已经下载并安装了导致这些错误的驱动程序的正确版本。我恐怕需要一点反复试验。

然后,您可以输入以下内容来检查您的TensorFlow版本:

c1ce3c5724b8874356952fb09be3281b23dad594

请注意,前后有两个下划线versionTensorFlow出其版本号,如下所示:

print(tf.__version__)

9f6c08dbb1fe7af245a0df66970089a61bb2377d

在有一个完整的TensorFlow运行时环境,在Windows上支持GPU

本文由阿里云云栖社区组织翻译。

文章原标题《installing-tensorflow-with-gpu-on-windows-10》,

作者: Laurence Moroney,译者:乌拉乌拉,审阅:袁虎。

文章为简译,更为详细的内容,请查看原文

相关实践学习
基于阿里云DeepGPU实例,用AI画唯美国风少女
本实验基于阿里云DeepGPU实例,使用aiacctorch加速stable-diffusion-webui,用AI画唯美国风少女,可提升性能至高至原性能的2.6倍。
相关文章
|
2月前
|
机器学习/深度学习 存储 PyTorch
【AMP实操】解放你的GPU运行内存!在pytorch中使用自动混合精度训练
【AMP实操】解放你的GPU运行内存!在pytorch中使用自动混合精度训练
67 0
|
3月前
|
并行计算 TensorFlow 算法框架/工具
win10上使用gpu版的tensorflow
win10上使用gpu版的tensorflow
|
4月前
|
并行计算 TensorFlow 算法框架/工具
TensorFlow识别GPU难道就这么难吗?还是我的GPU有问题?
TensorFlow识别GPU难道就这么难吗?还是我的GPU有问题?
90 0
|
4月前
|
TensorFlow 算法框架/工具 异构计算
Windows部署TensorFlow后识别GPU失败,原因是啥?
Windows部署TensorFlow后识别GPU失败,原因是啥?
48 0
|
3月前
|
机器学习/深度学习 算法 TensorFlow
文本分类识别Python+卷积神经网络算法+TensorFlow模型训练+Django可视化界面
文本分类识别Python+卷积神经网络算法+TensorFlow模型训练+Django可视化界面
57 0
文本分类识别Python+卷积神经网络算法+TensorFlow模型训练+Django可视化界面
|
4月前
|
机器学习/深度学习 异构计算 Python
Bert-vits2最终版Bert-vits2-2.3云端训练和推理(Colab免费GPU算力平台)
对于深度学习初学者来说,JupyterNoteBook的脚本运行形式显然更加友好,依托Python语言的跨平台特性,JupyterNoteBook既可以在本地线下环境运行,也可以在线上服务器上运行。GoogleColab作为免费GPU算力平台的执牛耳者,更是让JupyterNoteBook的脚本运行形式如虎添翼。 本次我们利用Bert-vits2的最终版Bert-vits2-v2.3和JupyterNoteBook的脚本来复刻生化危机6的人气角色艾达王(ada wong)。
Bert-vits2最终版Bert-vits2-2.3云端训练和推理(Colab免费GPU算力平台)
|
1天前
|
机器学习/深度学习 运维 监控
TensorFlow分布式训练:加速深度学习模型训练
【4月更文挑战第17天】TensorFlow分布式训练加速深度学习模型训练,通过数据并行和模型并行利用多机器资源,减少训练时间。优化策略包括配置计算资源、优化数据划分和减少通信开销。实际应用需关注调试监控、系统稳定性和容错性,以应对分布式训练挑战。
|
1天前
|
机器学习/深度学习 TensorFlow 调度
优化TensorFlow模型:超参数调整与训练技巧
【4月更文挑战第17天】本文探讨了如何优化TensorFlow模型的性能,重点介绍了超参数调整和训练技巧。超参数如学习率、批量大小和层数对模型性能至关重要。文章提到了三种超参数调整策略:网格搜索、随机搜索和贝叶斯优化。此外,还分享了训练技巧,包括学习率调度、早停、数据增强和正则化,这些都有助于防止过拟合并提高模型泛化能力。结合这些方法,可构建更高效、健壮的深度学习模型。
|
1天前
|
机器学习/深度学习 TensorFlow 算法框架/工具
TensorFlow入门指南:基础概念与安装
【4月更文挑战第17天】TensorFlow入门指南介绍了该流行深度学习框架的基础概念和安装步骤。核心概念包括张量(多维数组)、计算图(表示计算任务的图结构)、会话(执行环境)以及变量(存储模型参数)。安装TensorFlow可通过pip或conda,GPU支持需额外条件。安装成功后,通过Python验证版本即可开始使用。
|
2月前
|
机器学习/深度学习 人工智能 API
人工智能应用工程师技能提升系列2、——TensorFlow2——keras高级API训练神经网络模型
人工智能应用工程师技能提升系列2、——TensorFlow2——keras高级API训练神经网络模型
31 0