使用 Anaconda 安装 TensorFlow-gpu

Question

这个周末我一直在尝试安装 Tensorflow with GPU support 以在我的计算机上工作，但是我在使用 pip/conda 方面不是很有经验，现在看完后很困惑并尝试了很多来自网络的不同教程/方法。

我有一个 GeForce GTX 1650 显卡，并且我已经安装了 Cuda 10.0（也是 11.2，但我从“PATH”中删除了它并且只使用了 10.0版本，我认为这不是问题）。

我已下载 cuDNN 7.5.0 for CUDA 10，我认为我已正确复制和放置文件（已安装 cuDNN）。

我只是想让一些版本的 Tensorflow-gpu 工作，但你可以在图像上看到我现在一直在尝试的 Tensorflow 版本。

我尝试从我的计算机上安装和卸载 Python（我也多次重新安装 Anaconda），因为我不确定如果我安装一个 Python 版本（在我的系统上）是否需要我的 Anaconda 环境中的 Python 版本（在我的示例中为 Python 3.7）。

有谁知道如何使用我的设置（cuDNN 7.5.0、CUDA 10）在 Windows 10 上安装 Tensorflow GPU，或者可能在 Python 版本或类似于我的 Anaconda 问题上遇到了一些问题？

Answer 1

按照这些步骤在 Windows 系统上安装 Tensorflow GPU。

1. 确保安装了正确版本的 Visual Studio。检查here。

2. 按照here 中提到的说明为 Windows 系统设置 CUDA

3. 安装 TensorFlow

#check current python version
python --version
#Create the virtual environment
conda create -n tf python=PYTHON_VERSION 
#Activate the tf environment
conda activate tf
#Install Tensorflow
pip install tensorflow
#Install CUDA and cuDNN using conda and make sure CUDA and cuDNN version should match the Tensorflow version
conda install -c anaconda cudatoolkit=10.0 cudnn=7.5