-2

我是 tensorflow 的新手，需要一些帮助来安装它。

我有带有 python 3.7.7 的 Miniconda3。

我有8GB RAM，i7-7700hq 核心和 GTX-1050，4GB RAM 显卡，带有Cuda 驱动程序版本 9.2.209版本（我从 NVCUDA.DLL 中的 Nvidia 控制面板看到这个）

我试图按照媒体、youtube 上的说明进行操作，但没有用。

有人可以帮我在我的笔记本电脑上设置 tensorflow-gpu（最好是 2.1.0，但我会对任何版本感到满意，只要它能正常工作）吗？

请随时提出您可能需要回答的任何问题/信息。

谢谢 ：）

4

1 回答 1

2

版本检查-

  • 检查您使用的 GPU (win+X) -> 设备管理器 -> 显示适配器
  • 确定您要使用的 Tensorflow 版本。
  • 从这里确定 tensorflow 版本支持的 CUDA 版本。2.1.0 支持 10.1

先决条件-

  • 启用了 CUDA 的 GPU（点击此处查看）
  • 视觉工作室。这是安装 GCC 编译器所必需的。您也可以使用 Windows SDK，但我没有尝试过。

步骤-

  1. 在此处下载所需版本的 CUDA 工具包
  2. 在此处为您的 cuda 版本下载最新的 cuDNN （您必须为此向 Nvidia 注册）。
  3. 安装 CUDA 工具包。它也会自动安装所需的 Nvidia 驱动程序。
  4. 使用这些说明安装 cuDNN

编辑- 您需要确保您的显卡具有必要的驱动程序以支持 tensorflow 所需的 CUDA 版本。我必须阅读每个驱动程序的文档才能做到这一点。（检查您的 GPU 的驱动程序）

于 2020-04-13T04:32:22.830 回答