2020年7月

背景

Python 3.7 Windows10 64

步骤

  1. 安装Tensorflow库 目前版本是2.2

    pip install tensorflow
    pip install tensorflow-gpu (二选一,两个在使用上等价,只是性能有差别,gpu会稍微快一点)

  2. 安装CUDA显卡加速软件(GPU的并行计算框架) tensorflow的支持版本是 10.1

    下载地址: https://developer.nvidia.com/cuda-10.1-download-archive-update2

  3. 安装CUDNN(神经网络的加速包) 下载7.6即可

    下载地址: https://developer.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.1_20191031/cudnn-10.1-windows10-x64-v7.6.5.32.zip

- 阅读剩余部分 -