当前位置: 首页 > 面试题库 >

我如何使用TensorFlow GPU?

步德宇
2023-03-14
问题内容

如何在Python 3.6 x64中使用 TensorFlow GPU 版本而不是 CPU 版本?

import tensorflow as tf

Python正在使用我的 CPU 进行计算。
我可以注意到它,因为我有一个错误:

您的CPU支持该TensorFlow二进制文件未编译为使用的指令:AVX2

我已经安装了tensorflow和tensorflow-gpu。

如何切换到GPU版本?


问题答案:

遵循本教程Tensorflow GPU,我做到了,它运行完美。

注意! -安装 9.0版! Tensorflow-gpu不支持较新的版本

脚步:

  1. 卸载旧的tensorflow
  2. 安装tensorflow-gpu pip install tensorflow-gpu
  3. 安装Nvidia图形卡和驱动程序(您可能已经拥有)
  4. 下载并安装CUDA
  5. 下载并安装cuDNN
  6. 通过简单程序验证

from tensorflow.python.client import device_lib print(device_lib.list_local_devices())



 类似资料:
  • 问题内容: 你们能用EPOLLHUP为我提供一个很好的示例代码来处理死者吗?我知道这是检测到用户断开连接的信号,但不确定如何在代码中使用它。 问题答案: 您没有检测到同级关闭(这表明套接字意外关闭,即通常是内部错误)。 使用它非常简单,只需将标志与您要赋予的其他标志“或”即可。因此,例如,代替write 。 之后,如果另一端关闭了连接(可能要关闭套接字),请执行随后要执行的操作。 请注意,从套接字

  • 在一个flink项目中,我使用一个case类click。 这个类填充了数据集,并且在日期为Java8的情况下可以很好地工作。在Java7环境中切换到org.joda(Version2.9)之后,对数据集中的click对象的调用不像以前那样执行。对click对象的date字段的某些函数的访问引发。这些函数的例子有等。我能够确保click类的日期字段不为空。我怀疑joda时间库与kryo序列化的交互不

  • 问题内容: 有什么好的资源可以开始使用Node.JS?有没有好的教程,博客或书籍? 当然,我已经访问了它的官方网站http://nodejs.org/,但是我认为他们拥有的文档不是一个很好的起点。 问题答案: 讲解 NodeSchool.io交互式课程 节点的艺术(Node.js简介) 你好,世界 Node.js指南 使用Node.js,Express和MongoDB构建博客 适用于初学者的Nod

  • 问题内容: 我的一部分代码获得了OuterHTML属性 因此我可以进行涉及解析的内容。 虽然在Firefox上的javascript中没有OuterHTML属性,但我找不到其他方法来获取此字符串。有想法吗? 问题答案: 弄清楚了! getAttribute不起作用,但是getAttributeNode很好; D

  • 问题内容: 我正在尝试在自定义身份验证中覆盖is_authenticated。我有一个简单的东西(开始)是这样的: 然后在我的ModelResource中 我不断收到500错误。我在数据库中只有一个用户名,这是我要验证的用户。 关于它为什么返回此错误的任何想法?我将如何允许api客户端登录? 谢谢您的帮助。 问题答案: 您的方法将尝试使用您要进行身份验证的用户名创建一个新用户。正如您所注意到的,这