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要快太多太多了

     
  • 相关阅读:
    笔试面试题集锦
    Mosquitto pub/sub服务实现代码浅析-主体框架
    查找算法(一)
    基数排序
    插入排序-----希尔排序
    插入排序------直接插入排序
    归并排序
    选择排序--------简单选择排序
    交换排序------冒泡排序
    排序算法
  • 原文地址:https://www.cnblogs.com/Renyi-Fan/p/13457402.html
Copyright © 2011-2022 走看看