前言
买了新的笔电并成功配置了Win10上面的tensorflow的GPU版
毕竟稍微大一点的神经网路CPU跑起来实在慢的要人命!
纪录了一些流程希望帮助也在安装的朋友省一点时间
可以赶快开始利用新买的显卡来工作或做程式学习
配置流程
首先下载并安装Anaconda+Python3.7的版本
https://www.anaconda.com/distribution/
下载后直接参考tensorflow官网的说明
不要相信来路不明的教学啦QQ乖乖照着TensorFlow-gpu官网做才是最准的!!!
我们需要做的事情有:1
2
31. 确认GPU驱动有更新
2. CUDA@Tookit 10.0安装
3. cuDNN SDK 7.5(forCUDA 10.0)安装
官方网址:https://www.tensorflow.org/install/gpu
配置方案,可以参考下图
照着上面的标示安装必要程序
1.显卡驱动因为刚买来还满新的进Geforce Experience更新一下不用重载
根据显卡驱动,选择对应的CUDA版本号
查看本机的显卡驱动版本号,方法如下:
WIN+R,输入CMD,然后输入”nvidia-smi”,即可显示对应的显卡驱动版本号,有些还会显示对应的支持CUDA的版本号,本机验证CUDA支持10.0。
2.下载CUDA 10.0 + CUDNN7.5 for 10.0 (版本记得要一样喔不然会错误)
- 1.下载CUDA 10.0, CUDA Toolkit Archive
如想了解是否正确安装可以1
nvcc - V
如果成功应该是如下图所示:
- 2.下载CUDNN
&emspp;直接去NVDIA官网去下载就好啦~大家想要下载,还需要注册NVDIA账户,很麻烦,不过下载好像可以直接微信登录了,我没试过,大家就直接微信登录就好啦,下面附上链接:下载链接这里面有好几个for CUDA10.0的,大家一定要注意,因为关系到我们后面tensorflow的版本,这里出错了,后面就会报错!!! 这里我们选择cuDNN v7.5的版本
到这里下载完成!完成后咱们开始解压,然后将相应的包,放到cuda相应包底下。
咱们只需要拿出这些文件夹里面文件放到想要cuda文件夹即可,我的cuda文件夹地址为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
供大家参考。
- TensorFlow-GPU下载和安装
我们可以通过进入英伟达控制面板,点击帮助,选择系统信息,再点组件,看到我们的RTX 2060显卡是支持CUDA10。
前面我让大家下载了CUDA9.2,没坑大家哦,CUDA9.2也是支持RTX 2060的。tensoeflow官方现在无论哪个版本都不支持,所以我们还是去求助Github!!!
安装1
2// An highlighted block
pip install tensorflow_gpu
测试
1 | // An highlighted block |
结果
已经能够识别出RTX2060显卡了,并成功输出!到这里也算完成了,大家安装遇到的坑也欢迎来交流!