大致思路
1 gpu的型号和驱动版本是所有约束的起点,然后依次是 cuda,cudnn 和 tensorflow-gpu 和python
2 cuda cudnn tensorflow-gpu 之间的兼容关系一定要严格按照表格来,不要有侥幸心理!
3 anaconda 只是创建了一个环境,这个环境是空的,在这个环境里得重新安装各种包,甚至是anaconda自己的包
4 python 不要贪最新的release,先把1中各项表查好 再选python版本
步骤
桌面右键 NVIDIA 控制面板,点击左下角系统信息,然后看驱动程序版本

查看cuda 和 驱动的对应关系,网上有些帖子可能会点击组件,然后发现有cuda版本了。组件里的cuda版本不要管。查cuda和显卡驱动关系

发现cuda的版本大概在 10.0-10.2之间
去tensorflow的官网查询 cuda cudnn tensorflow-gpu三者版本兼容关系
tensorflow官网查依赖

所以 cuda 版本选10 cudnn 选7.4 tensorflow-gpu 选择1.14
然后找到nvidia的官网 下载 cuda 和cudnn
cuda下载地址


一开始我选择的是10.2 但是装到最后测试的时候报错,日志提示安装10.0,所以选了10.0
这也侧面反映tensorflow官网的版本精度不够
cudnn下载地址

cudnn下载前得在Nvidia的网站上注册一下,才能下载。

安装cuda

选择自定义安装
一路选是,同时记住安装路径

本文介绍了如何在Win10系统,拥有RTX 2060 Super显卡的环境下,使用Anaconda搭建TensorFlow-GPU 1.14、CUDA 10.0和CUDNN 7.6.4的机器学习环境。关键步骤包括检查GPU驱动、确定兼容的CUDA和CUDNN版本、下载并安装CUDA和CUDNN、配置环境变量,以及创建和激活Anaconda虚拟环境,最后验证GPU是否能被TensorFlow识别。
最低0.47元/天 解锁文章
1万+

被折叠的 条评论
为什么被折叠?



