Windows7系统安装配置TensorFlow-GPU运算环境

1. 概述

        在深度学习实践中,对于简单的模型和相对较小的数据集,我们可以使用CPU完成建模过程。例如在MNIST数据集上进行手写数字识别,我们可以使用CPU来完成,采用经典的LeNet-5模型只需要十几分钟就能完成(具体取决于电脑配置)。

        然而,为了达到更好的效果或者构建更复杂的模型,一般需要更大的数据集,例如经典的ImageNet数据集。假如使用CPU训练模型将显得无比吃力(将会花费很长长长长长时间),此时GPU就可以派上用场了,利用其强大的并行计算能力训练模型会变快很多。

本次教程主要分为以下几个步骤:

  • 安装显卡驱动
  • 安装CUDA
  • 安装CuDNN
  • 安装TensorFlow-GPU

注意:请确保已安装python(若未安装请参考其它教程),并且最好为python3.5或3.6版本。

由于NVIDIA网站有时下载速度慢,提供百度网盘下载链接

  cuda_9.0.176_w

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值