我一直在尝试使用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 install
tensorflow GPU。你需要设置运行Tensorflow GPU所需的配置,就像你第一次在机器上安装python一样。您需要安装CUDA工具包和许多其他东西。第一次安装它可能是一个繁重的过程,所以一定要遵循一些指南,这会为你省去很多麻烦。一旦设置完成,我建议使用anaconda安装tensorflow gpu。这样,您也可以指定版本。
遵循本教程Tensorflow GPU。我做到了,而且效果很好。
YouTube上还有很多其他教程,你可以继续学习。其中一些已经在评论部分突出显示。