win10安装tensorflow-gpu2.0.0 keras CUDA10.0 cuDNN7.6.0_Apollo爱搞学习-CSDN博客 - 图1

    ApollooxX 2020-04-03 13:18:53 win10安装tensorflow-gpu2.0.0 keras CUDA10.0 cuDNN7.6.0_Apollo爱搞学习-CSDN博客 - 图2
    426 win10安装tensorflow-gpu2.0.0 keras CUDA10.0 cuDNN7.6.0_Apollo爱搞学习-CSDN博客 - 图3
    收藏 1

    最后发布: 2020-04-03 13:18:53 首次发布: 2020-04-03 13:18:53

    版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

    win10+tensorflow-gpu2.0.0+keras+cuda10.0.130(不可以是 10.1)+7.6.0(要大于 7.4.1)

    前提:电脑已经安装了 Anaconda3。

    声明:这个版本的 cuda 和 cudnn 已经不需要手动下载安装了,直接命令行通过 conda 即可安装。

    额外:如果之前没有更改过 pip 和 conda 镜像源,参考这个:
    pip、conda 更改为国内源

    正文:
    1、首先检查电脑英伟达显卡驱动,键盘 win+r 输入 cmd 打开,输入 nvidia-smi, 即可看到驱动版本号,需要大于 410.x。如果不够,到英伟达官网下载:https://www.nvidia.cn/Download/index.aspx?lang=cn

    2、安装 cuda+cudnn。进入自己的安装环境,我的直接安装在 默认的 base 环境下,打开开始菜单栏的 Anaconda Prompt (Anacanda3),默认是 base 环境下(命令行最前面小括号内就是当前环境名),依次输入:conda install cudatoolkit==10.0.130conda install cudnn==7.6.0, 安装前会问是否要安装更新,输入 y 即可可以下载安装。

    3、再安装 tensorflow-gpu2.0.0。输入 conda install tensorflow-gpu==2.0.0,即可开始安装。

    4、最后安装 keras,,输入 pip install keras 或者 conda install keras

    5、查看 tensorflow 以及 cuda 是否安装成功,输入 c:进入用户目录,输入 python进入 python 环境,输入 import tensorflow as tf,显示成功打开一个动态库,再输入 tf.__version__,显示 '2.0.0‘表示 TensorFlow2.0.0 安装成功。
    输入 tf.test.is_gpu_available(), 若返回 true 标志可使用 GPU。
    https://blog.csdn.net/qq_42902264/article/details/105291024