Win10安装tensorflow-gpu

本来准备使用虚拟机安装ubuntu来开启深度学习之旅的，但是装完发现vmware装的系统不支持GPU直通，任务一跑就cpu资源不足停掉了。平时使用的又是windows系统，装双系统很不方便，所以开始折腾windows的tensorflow安装。

目录

• 1 安装anaconda
  • 1.1 下载安装
  • 1.2修改源
  • 1.3 创建conda环境
• 2 安装CUDA和cuDNN
  • 2.1确认显卡支持的CUDA版本
  • 2.2下载CUDA和cuDNN
  • 2.3 安装CUDA
  • 2.4 安装cuDNN
  • 2.5 修改环境变量
• 3 安装tensorflow-gpu

1 安装anaconda

1.1 下载安装

地址： https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

1.2修改源

执行以下命令生成配置文件

conda config --set show_channel_urls yes	

修改用户目录下的 .condarc 文件

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

运行 conda clean -i 清除索引缓存，保证用的是镜像站提供的索引。

1.3 创建conda环境

cmd执行

conda create tensorflow python=3.8
activate tensorflow //激活环境
deactivate //退出环境

2 安装CUDA和cuDNN

2.1确认显卡支持的CUDA版本

桌面右键NVIDIA控制面板-帮助-系统信息-组件

显示的版本为11.0.228，那么11.0就是本机支持的最高CUDA版本，到tensorflow官网(https://www.tensorflow.org/install/source_windows?hl=zh-cn)查找匹配的tensorflow版本,如下图意味着我们要下载tensorflow-gpu-2.4.0和cuDNN8.0、CUDA11.0版本

2.2下载CUDA和cuDNN

在nvidia官网没找到相关的cuda历史版本下载页面，可能没仔细找的原因，直接google一下 CUDA 11.0跳转到相关版本下载地址，如下

复制下载链接到迅雷下载(有cdn加速)

然后下载cuDNN(https://developer.nvidia.com/rdp/cudnn-archive) ，需要注册NVIDIA账号，找到适配的版本

2.3 安装CUDA

安装过程中注意 提示安装项目时，只安装toolkit就行，其他如显卡驱动、GeForce Experience等如果电脑已经安装过就取消勾选，避免降了版本。其他均按默认。完成后，cmd执行nvcc -V 验证是否安装成功。

2.4 安装cuDNN

解压压缩包，并提取bin、include、lib里的文件复制到CUDA的同名目录里面，如果是按默认地址，那应该是 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

2.5 修改环境变量

添加CUDA安装目录

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

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64

到环境变量PATH中，然后重启电脑。

3 安装tensorflow-gpu

cmd执行

activate tensorflow //激活之前创建的环境

查找 tensorflow版本

conda search tensorflow-gpu

如果有需要的对应版本直接执行

conda install tensorflow-gpu=2.4.0

我在执行过程中发现tuna源没有需要的2.4.0版本，只能另外找豆瓣的源进行下载

pip install tensorflow-gpu==2.4 -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

python
import tensorflow as tf
tf.config.list_physical_devices('GPU')

如果能列出显卡则大功告成