在 Windows 10 上为 GTX 1050 安装带有 conda 的 Tensorflow-GPU

Question

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

我有 Miniconda3 python 3.7.7.

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

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

谁能帮我在我的笔记本电脑上设置 tensorflow-gpu（最好是 2.1.0，但我对任何版本都满意，只要它能正常工作）？

如有任何您可能需要回答的问题，请随时提出 questions/information。

谢谢:)

Answer 1

版本检查-

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

先决条件-

• 启用 CUDA 的 GPU (check here to check)
• Visual Studio。这是安装 GCC 编译器所必需的。您也可以使用 Windows SDK，但我还没有尝试过。

步骤-

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

编辑 - 您需要确保您的显卡具有必要的驱动程序以支持 tensorflow 所需的 CUDA 版本。为此，我必须查看每个驱动程序的文档。 (Check drivers for your GPU)