Win10系统下使用anaconda在虚拟环境下搭建tensorflow-gpu版本安装CUDA及CUDNN

说明：如果下载较慢可以提前加载一下国内镜像

基础环境：

			Win10
            Anaconda3

需要搭建的环境：

      CUDA 10.1

      CUDNN 7.6.5

      python3.7

      tensorflow-gpu 2.2.0

本文我们将通过anaconda创建的虚拟环境，即可安装任意版本的cuda。接下来是具体步骤。

1. 操作步骤

1)查看系统环境cuda版本

使用nvcc -V 可以查看系统环境下的cuda版本。

nvcc -V

2)查看版本支持

以管理员身份打开Anaconda Prompt,输入下列命令查看conda目前支持的cudatoolkit版本。(注cudatoolkit也是cuda)

conda search cudatoolkit --info

可以看到conda支持cuda10.1，接下来看看支持cudnn 的版本。

conda search cudnn --info

也看到了我所需要的版本了，接下来介绍安装了。

3)创建虚拟环境

创建一个名为tensorflow的虚拟环境，python为3.7版本。

conda create -n tencorflow python=3.7

然后激活环境，如文本所示。

conda activate tensorflow

4)安装cuda

conda install cudatoolkit=10.1

5)安装cudnn

conda install cudnn=7.6.5

6)安装tensorflow_gpu == 2.2.0

pip install tensorflow-gpu==2.2.0

2. 测试

1.用conda list看下此环境已经安装的包

conda list

此时说明已经安装好。

2. 用demo测试下

在终端输入python进入python环境。

输入import tensorflow as tf

import tensorflow as tf

到此，tensorflow-gpu基本环境搭建完成，可能还会遇到一些包无法找到，读者按照错误提示，在Anaconda Prompt中安装即可。

版本可能有更新大家按照所需安装即可。