如何使用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