如何在windows系统上的安装TensorFlow-gpu,全套方法在这拿走

看到有很多朋友问TensorFlow-gpu如何配置,我打算分享下我的安装经验。

first and first

兄dei,你的显卡是NVIDIA 1060 以上吗?
不是的话,先去买个深度学习 穷人入门版 的1060的显卡吧(最低1050ti才可以在深度学习领域划水),在公司工作的、土豪建议直接2080Ti,或者专业图形显卡P100

只有NVIDIA显卡支持GPU加速

TITAN系列显卡

在这里插入图片描述

专业图形显卡

在这里插入图片描述

首先看看你的系统配置,WIN10系统上的鼠标右键打开NVIDIA控制面板,
在这里插入图片描述
找到“组件”里的CUDA型号,比如我的就CUDA10.2
在这里插入图片描述

安装TensorFlow-gpu

到如下网址查看你的匹配的TensorFlow-gpu的版本
https://www.tensorflow.org/install/source_windows
在这里插入图片描述
在你的虚拟环境里(最好用anaconda创建python=3.6虚拟环境,python3.7太新新,有些git代码不支持,python3.5太老,不知道如何创建anaconda虚拟环境的百度),输入如下命令,记得选择对应的TensorFlow-gpu版本安装

pip install tensorflow-gpu==1.14 -i http://pypi.douban.com/simple

在这里插入图片描述

准备CUDA和cuDNN安装文件

打开下面的网址下载,符合你电脑的CUDA
https://developer.nvidia.com/cuda-toolkit-archive
在这里插入图片描述
在这里插入图片描述

进入一下网站https://developer.nvidia.com/rdp/cudnn-download 下载相配的cuDnn 版本
进这个网要注册登录一下,
在这里插入图片描述
才能看到如下界面,图中有写明cuDNN 几点几对应 CUDA哪个版本。注意选取下载。
在这里插入图片描述
在这里插入图片描述
现在有了CUDA和cuDNN两个必备的文件,接下来先双击第一个文件cuda.exe进行安装
在这里插入图片描述

CUDA安装

安装路径可以默认也可以自定义。之后点击ok就行
在这里插入图片描述
等一会,加载。。。
在这里插入图片描述
keep waiting(好事多磨,再等会。。。)
在这里插入图片描述
点同意意,继续
在这里插入图片描述
如果不知道怎么选,就选择精简安装。
在这里插入图片描述
选默认或自定义,记下安装路径
在这里插入图片描述
最终“下一步”,然后“完成”就行。
配置系统环境变量,选择path:
在这里插入图片描述
以下两个是自动就有的,没有自己加一下。
在这里插入图片描述
win+R打开cmd,验证安装:nvcc -V
出现如下图代码,就是成功安装CUDA了
在这里插入图片描述

配置cuDNN

接下来,该用到第二个文件
在这里插入图片描述
解压后出现bin、include、lib三个文件夹
在这里插入图片描述
一个一个复制粘贴到刚才的CUDA安装路径下
在这里插入图片描述
接下来添加cudnn的path
这里特别重要,改成你自己的路径

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include
在这里插入图片描述

至此cuDNN安装完成!!

测试

1、进入cmd激活你装好TensorFlow-gpu环境,键入python进入python shell;
2、输入import tensorflow as tf导入tensorflow库,无报错即成功安装TensorFlow;
3、键入代码

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()

有如下图的效果,即说明使用了GPU进行计算。
在这里插入图片描述

提醒

pycharm里运行TensorFlow-gpu会报一些warning,那些提醒是因为版本太新或者有些稍微不匹配的问题,不过不用担心,只要会出结果,能运行就是成功了。

tensorflow-gpu安装成功!

好用点个赞呗!本人持续更新python人工智能方面的东西。大家一起进步!

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

宇疏桐

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值