win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程

安装环境:win10+Anaconda4.8.3 + Python3.7.7
安装版本:tensorflow1.14.0+CUDA10.0+cuDNN7.6.5

Anaconda安装

Anaconda是一个开源的Python发行版本,其包含了Python、conda等180多个常用科学计算包及其依赖项。相比单独安装Python主程序,安装Anaconda后不需要再单独安装大部分的第三方库。
Anaconda官网下载地址:https://www.anaconda.com/

Tensorflow-GPU版本安装

运行Anconda Prompt进入命令行

  1. 检查Anaconda的版本:conda --version
  2. 检查Anaconda中安装的环境变量:conda info --envs
    在这里插入图片描述
  3. 检查当前可使用的Python版本:conda search --full-name python
  4. 创建Anaconda内置的Python版本解析器,其实就是添加tensorflow环境:
    安装GPU版本:conda install --name tensorflow-gpu python=3.7
    安装CPU版本:conda install --name tensorflow python=3.7
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第1张图片
  5. 激活tensorflow环境:activate tensoflow
  6. 检查Python的版本:
    在这里插入图片描述
  7. 退出当前环境:conda deactivate
    在这里插入图片描述
  8. 检查当前Anaconda创建的环境:conda info --envs
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第2张图片
    (如果需要删除Anconda中创建的环境:conda remove --name tensorflow-gpu --all
  9. 查询可供安装的tensorflow版本:conda search --full-name tensorflow
  10. 安装tensoflow-GPU版本:conda install tensorflow-gpu==1.14.0(版本可自选,我安装的是v1.14.0)
    (也可以安装完CUDA和cuDNN后再安装tensorflow)
    安装tensorflow-CPU版本:conda install tensorflow==1.14.0

CUDA和cuDNN版本选择

  1. 查看本设备的CUDA驱动适配版本
    进入英伟达控制面板,点击帮助->系统信息,可以查看GPU的完整版本信息
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第3张图片
    点击组件,可以查看本设备GPU所支持的CUDA版本(可看到本机支持CUDA10.2及以下版本)
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第4张图片
  2. 查看tensorflow适配的CUDA版本
    在tensorflow官网https://www.tensorflow.org/install/source_windows#gpu可以查询tensoflow各版本对应的CUDA和cuDNN版本,切记版本对应安装!!!(比如安装的tensorflow-gpu版本是v1.14.0,那么CUDA应该装v10.0,cuDNN应该装v7.6.5)
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第5张图片

CUDA下载安装

CUDA是NVIDIA推出的只能用于自家GPU的并行计算框架,只有安装该框架才能够进行复杂的并行计算。主流的深度学习框架也都是基于CUDA进行GPU并行加速的。
CUDA官网下载地址: https://developer.nvidia.com/cuda-downloads
win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第6张图片

  1. 推荐选择自定义安装
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第7张图片
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第8张图片
  2. 推荐默认安装路径
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第9张图片
  3. 检查CUDA是否安装成功以及版本信息:cmd输入nvcc --version或nvcc -V
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第10张图片

cuDNN下载安装

cuDNN是针对深度卷积神经网络的GPU加速库。
cuDNN官网下载地址:https://developer.nvidia.com/cudnn
注意:必须选择和你安装的CUDA相匹配的版本!!!
win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第11张图片

  1. 把下载的cudnn-10.0-windows10-x64-v7.6.5.32_2.zip压缩包解压
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第12张图片
  2. 把上面3个文件夹复制到CUDA安装目录下面(直接拖拽粘贴即可,没有重复的文件)
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第13张图片
  3. 安装完成后,进入系统环境变量中添加环境变量
    此电脑->属性->高级系统设置->环境变量->系统变量->选择Path进入编辑
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第14张图片
    添加如下图两个新的环境变量(所以一定要记住CUDA的安装路径!!!v10.0为CUDA版本号,根据下载的版本进行更改)
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第15张图片
  4. 检查安装结果
    win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第16张图片

Hello,Tensorflow!

在Anaconda的Jupyter Notebook中运行Helloworld程序:
win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程_第17张图片

你可能感兴趣的:(win10系统下CUDA+cuDNN+tensorflow-GPU的详细安装流程)