gpt4 book ai didi

python - Tensorflow-gpu 仅使用 1.14 版本训练自定义对象检测模型

转载 作者:行者123 更新时间:2023-11-30 08:44:55 24 4
gpt4 key购买 nike

我很高兴找到了一个可以训练和测试自定义对象检测模型的设置,但我很困惑为什么最新的tensorflow-gpu版本不能在我的电脑上运行,而另一个设置可以在我的笔记本电脑上运行:

我正在使用的设置(PC):

我正在使用的设置(笔记本电脑):

如何(何时?)在我的 PC 上使用 tensorflow-gpu v2.0.0 训练自定义模型?
谢谢

最佳答案

如果您需要在Windows上运行tensorflow-gpu,Microsoft Visual Studio是一个先决条件。这个要求也提到了here 。您可以下载Visual Studio 2019社区版here

虽然tensorflow-gpu installation docs中没有直接提到这个要求,它是 CUDA installation guide for Windows 的一部分。此外,您还需要确保拥有具有 CUDA 计算能力3.5 或更高的 NVIDIA GPU 卡。您可以检查您的 GPU 是否满足计算能力要求 here .

以下是一些可帮助安装 tf-gpu 的资源:

  1. https://towardsdatascience.com/installing-tensorflow-with-cuda-cudnn-and-gpu-support-on-windows-10-60693e46e781
  2. https://www.youtube.com/watch?v=KZFn0dvPZUQ

关于python - Tensorflow-gpu 仅使用 1.14 版本训练自定义对象检测模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59594863/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com