如何在Windows 10上为TensorFlow使用python中的CUDA设置GPU?

如何在Windows 10上为TensorFlow使用python中的CUDA设置GPU?,python,tensorflow,Python,Tensorflow,我一直在尝试使用TensorFlow,但每当我尝试运行某些东西时,我都会收到一个错误,说我需要在我的机器上设置一个GPU。我听说我需要下载一些CUDA附件来运行它,但不太确定 2020-11-18 11:23:57.579846: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. 我有一

我一直在尝试使用TensorFlow,但每当我尝试运行某些东西时,我都会收到一个错误,说我需要在我的机器上设置一个GPU。我听说我需要下载一些CUDA附件来运行它,但不太确定

2020-11-18 11:23:57.579846: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

我有一个戴尔xps 15 9570,所以我有一个NVIDIA GeForce RTX 1050 Ti GPU。我认为这足够硬件运行TensorFlow,但我不太确定。

我认为您当前的NVIDIA GTX 1050 Ti不会遇到任何问题

你需要使用TensorFlow GPU。您不能直接
pip安装
tensorflow-GPU。您需要设置运行Tensorflow GPU所需的配置,就像您第一次在机器上安装python一样。您需要安装CUDA工具包和许多其他东西。第一次安装它可能是一个繁重的过程,因此请确保您遵循一些指南,这将为您节省很多麻烦。一旦设置完成,我建议使用anaconda安装tensorflow gpu。这样,您也可以指定版本

遵循本教程。我做到了,效果非常好


YouTube上还有很多其他的教程,你可以继续学习。其中一些已经在评论部分突出显示。

您安装了CUDA运行库吗?我不这么认为。我可以安装它们吗?你需要安装tensorflow gpu。按照这个系统,根据你的系统规格做视频的修改,它检查你的图形卡是否被列入英伟达CUDA启用的产品列表。我检查了英伟达RTX 1050,但不是RTX 1050 TI。这是否意味着它不能在我的设备上运行。网站链接:RTX 1050 Ti支持CUDA,这不应该成为问题。如果您尚未安装CUDA运行时,则无法通过pip添加,它需要是系统级安装。请尝试按照处的指南进行操作,并替换相关版本以反映您的tensorflow安装和系统。