如何在带有conda的windows 10上安装tensorflow 2 gpu？

Question

我有Nvidia RTX 2070，想要使用GPU编程。我已经通过Conda简单地安装了Cuda，但是tensorflow不知何故没有检测到任何GPU设备。我的tensorflow版本是2.3.0。

我已经尝试在anaconda提示符下使用以下命令，但没有成功。

conda create --name tfgpu python=3.8.5 tensorflow-gpu

我还尝试了下载CUDA工具包和cudnn文件的较长版本，并复制粘贴这些文件。但似乎什么都不起作用。当我运行以下命令时，我总是得到0。

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

真的需要帮助。

Answer 1

您可以查看本指南以了解TensorFlow的安装和测试，也请确保从Nvidia的官方网站下载并安装最新的cudnn和Cuda库。https://towardsdatascience.com/guide-to-conda-for-tensorflow-and-pytorch-db69585e32b8