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

时间:2020-04-12 15:29:43

标签: python tensorflow conda user-guide

我是tensorflow的新手,希望获得一些帮助来安装它。

我有带Python 3.7.7的Miniconda3。

我有 8GB RAM,i7-7700hq内核和GTX-1050 ,具有 Cuda驱动程序9.2.209 版本的4GB RAM图形卡(我从Nvidia控件中看到了NVCUDA.DLL中的面板)

我尝试按照YouTube上的说明进行操作,但没有用。

有人可以帮我在笔记本电脑上设置tensorflow-gpu(最好是2.1.0,但只要能运行任何版本,我都会感到满意)?

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

谢谢:)

1 个答案:

答案 0 :(得分:1)

版本检查-

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

先决条件-

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

步骤-

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

修改- 您需要确保您的显卡具有必要的驱动程序来支持tensorflow所需的CUDA版本。我必须仔细阅读每个驱动程序的文档。 (Check drivers for your GPU