Win10 + RTX 2060 + tensorflow GPU

前言

买了新的笔电并成功配置了Win10上面的tensorflow的GPU版

毕竟稍微大一点的神经网路CPU跑起来实在慢的要人命!

纪录了一些流程希望帮助也在安装的朋友省一点时间

可以赶快开始利用新买的显卡来工作或做程式学习

配置流程

首先下载并安装Anaconda+Python3.7的版本

https://www.anaconda.com/distribution/

下载后直接参考tensorflow官网的说明

不要相信来路不明的教学啦QQ乖乖照着TensorFlow-gpu官网做才是最准的!!!
我们需要做的事情有:

1
2
3
1. 确认GPU驱动有更新
2. CUDA@Tookit 10.0安装
3. cuDNN SDK 7.5(forCUDA 10.0)安装

官方网址:https://www.tensorflow.org/install/gpu
tensorflow官网的GPU配置要求
配置方案,可以参考下图
tensorflow官网的GPU配置要求
照着上面的标示安装必要程序
1.显卡驱动因为刚买来还满新的进Geforce Experience更新一下不用重载
根据显卡驱动,选择对应的CUDA版本号
tensorflow官网的GPU配置要求

查看本机的显卡驱动版本号,方法如下:
  WIN+R,输入CMD,然后输入”nvidia-smi”,即可显示对应的显卡驱动版本号,有些还会显示对应的支持CUDA的版本号,本机验证CUDA支持10.0。

2.下载CUDA 10.0 + CUDNN7.5 for 10.0 (版本记得要一样喔不然会错误)

如想了解是否正确安装可以

1
nvcc - V

如果成功应该是如下图所示:
测试cuda是否安装成功-别人的图

  • 2.下载CUDNN
     &emspp;直接去NVDIA官网去下载就好啦~大家想要下载,还需要注册NVDIA账户,很麻烦,不过下载好像可以直接微信登录了,我没试过,大家就直接微信登录就好啦,下面附上链接:下载链接
    这里面有好几个for CUDA10.0的,大家一定要注意,因为关系到我们后面tensorflow的版本,这里出错了,后面就会报错!!! 这里我们选择cuDNN v7.5的版本
    到这里下载完成!完成后咱们开始解压,然后将相应的包,放到cuda相应包底下。
    测试cuda是否安装成功-别人的图
    咱们只需要拿出这些文件夹里面文件放到想要cuda文件夹即可,我的cuda文件夹地址为:
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
    供大家参考。
  1. TensorFlow-GPU下载和安装
    测试cuda是否安装成功-别人的图
    我们可以通过进入英伟达控制面板,点击帮助,选择系统信息,再点组件,看到我们的RTX 2060显卡是支持CUDA10。
    前面我让大家下载了CUDA9.2,没坑大家哦,CUDA9.2也是支持RTX 2060的。tensoeflow官方现在无论哪个版本都不支持,所以我们还是去求助Github!!!

安装

1
2
// An highlighted block
pip install tensorflow_gpu

测试

1
2
3
4
5
// An highlighted block
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

结果

测试cuda是否安装成功-别人的图
测试cuda是否安装成功-别人的图
已经能够识别出RTX2060显卡了,并成功输出!到这里也算完成了,大家安装遇到的坑也欢迎来交流!

常用网址

  1. CUDA Toolkit Archive
  2. cuDNN Archive
  3. Tensorflow prebuilt binary for Windows
  4. TensorFlow-gpu官网

参考文献

------------------------ The End ------------------------
0%