Windows+Python3.7+Tensorflow2+CUDA+cudnn安装详解

目录

 anaconda 

镜像源下载：

官网下载：

安装：

 切换镜像源：

​ CUDA+cudnn

查看计算机支持版本

​安装VS2019

 安装CUDA

安装cudnn

Tensorflow

创建环境

安装Tensorflow

---

 anaconda 

镜像源下载：

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

官网下载：

Anaconda | Individual Edition

安装：

一直默认，这一步选择：

切换镜像源：

如果是官网下载的anaconda，之后安装/更新一些包速度比较慢，所以建议切换镜像源。

1、打开anaconda prompt：

 2、输入命令：

conda config --set show_channel_urls yes

3、在user目录下找到.condarc文件，用记事本打开，将其内容换为：

channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirror.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirror.tuna.tsinghua.edu.cn/anaconda/cloud

 4、在anaconda prompt中依次输入：

conda clean -i

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

5、输入命令行查看安装结果：

conda info

CUDA+cudnn

tensorflowGPU 的软件/硬件需求可以在下面链接查到：

GPU support  |  TensorFlow (google.cn)

查看计算机支持版本

1、鼠标放置桌面空白处点击右键：

 2、点击左下角系统信息：

 3、我的CUDA版本11.3，查看对应python和tensorflow版本：

安装VS2019

 没有VS无法安装CUDA，已经安装的可以跳过这一步。

Downloads - Visual Studio Subscriptions Portal

需要先登录Microsoft账户，根据需要选择组件，我装在D盘了，下载安装会花一些时间。

安装CUDA

CUDA Toolkit Archive | NVIDIA Developer

点击上链接查找CUDA版本，我下载CUDA11.2.2(local)，network版安装太慢了。

打开安装文件，我选择自定义，也可以按照默认的安装。

安装cudnn

cuDNN Archive | NVIDIA Developer

1、点上链接需要登录NVIDIA账户填问卷，我下载的cudnn8.1.1

 2、解压后将文件里的bin|lib|include文件夹复制到以下路径（即CUDA安装路径）：

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

Tensorflow

创建环境

1、打开 anaconda prompt，输入下面命令创建环境（TF2为环境名，可以自定义）

conda create -n TF2 python=3.7

2、进入环境

conda activate TF2

安装Tensorflow

pip install tensorflow

自动安装适合python3.7的最新版tensorflow，我的是2.7.0

安装结束后依次输入以下命令查看是否成功：

# 进入TF2环境
conda activate TF2
# 导入tensorflow包
python
import tensorflow as tf
# 查看tensorflow版本
tf.__version__
# 查看GPU
tf.test.is_gpu_available()

安装成功！