zoukankan      html  css  js  c++  java
  • tensorflow2的gpu的版本安装(一些核心点)

    tensorflow2的gpu的版本安装(一些核心点)

    一、总结

    一句话总结:

    特别注意:当配置CUDA、cuDNN、tensorflow时,要确保这三者之间的版本对应一致

    二、tensorflow2的gpu的版本安装

    博客对应课程的视频位置:

    我没有详细介绍的部分,请大家自己去找度娘

    1、安装cuda

    各个版本位置

    https://developer.nvidia.com/cuda-toolkit-archive

     

    2、安装cuda对应版本的cuDNN

    cuDNN各个版本的下载网址:

    https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b

    这个下载要注册,太太太麻烦,大家可以去看下面这篇博文,亲测有效

    绕过Nvidia官方要求的注册或登陆步骤直接下载cuDNN的方法_weixin_40392957的博客-CSDN博客_cudnn下载
    https://blog.csdn.net/weixin_40392957/article/details/80207366

    3、配置系统变量

    4、测试是否安装成功

    配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe:
    首先win+R启动cmd,cd到安装目录下的 ...extrasdemo_suite,然后分别执行bandwidthTest.exedeviceQuery.exe,应该得到下图:

    如果以上两步都返回了Result=PASS,那么就算成功啦。

    5、tensorflow GPU测试tf.test.is_gpu_avaiable()返回false解决方法

    当配置CUDA、cuDNN、tensorflow时,要确保这三者之间的版本对应一致

    这个页面可以查看这三者版本的对应关系

    https://tensorflow.google.cn/install/source_windows

    6、当CUDA、cuDNN、tensorflow三者版本对应之后,gpu成功

    比如测试训练了一下卷积网络,GPU的确比CPU要快太多太多了

     
  • 相关阅读:
    Windows7下安装与破解IntelliJ IDEA2017
    Maven的目录结构和常用命令
    Maven三种仓库的配置
    Maven的安装
    Maven是什么
    SQL执行计划分析2
    SQL执行计划分析
    PHP 单点登录实现方案
    PHP消息队列实现及应用
    fastAdmin根据状态显示是否显示操作按钮
  • 原文地址:https://www.cnblogs.com/Renyi-Fan/p/13457402.html
Copyright © 2011-2022 走看看