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

Question

我对找到一个可以训练和测试自定义对象检测模型的设置感到欣喜不已，但是我很困惑为什么在笔记本电脑上可以使用其他设置时最新的tensorflow-gpu版本无法在我的PC上运行：< / p>

我正在使用的设置（PC）：

• 操作系统：Windows 10
• Python：3.6.4
• TensorFlow：1.14.0（gpu）
• CUDA：10
• cudnn：9.0吗？（不记得了）
• 回购+位置：https://github.com/tensorflow/models/research/object_detection
  
  

我正在使用的设置（笔记本电脑）：

• 操作系统：OSX
• Python：3.7.5
• TensorFlow：2.1.0-rc0（标准）
• 回购+位置：https://github.com/tensorflow/models/research/object_detection

如何（何时？）如何在PC上使用tensorflow-gpu v2.0.0训练自定义模型？

Answer 1

如果需要在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