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



我一直在尝试使用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 installtensorflow GPU。你需要设置运行Tensorflow GPU所需的配置,就像你第一次在机器上安装python一样。您需要安装CUDA工具包和许多其他东西。第一次安装它可能是一个繁重的过程,所以一定要遵循一些指南,这会为你省去很多麻烦。一旦设置完成,我建议使用anaconda安装tensorflow gpu。这样,您也可以指定版本。

遵循本教程Tensorflow GPU。我做到了,而且效果很好。

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

最新更新