windows7下使用GPU训练tensorflow深度学习模型

本文介绍了在Windows7系统中如何检查GPU支持,并详细讲解了安装CUDA、CUDNN的过程,以及配置环境变量的步骤。通过设置控制面板选择使用GPU的显卡,并在重启后通过测试代码验证GPU加速的TensorFlow深度学习模型是否成功运行。
摘要由CSDN通过智能技术生成

windows7先看一下自己有没有GPU,计算机->属性->设备管理器->显示适配器看到自己的显卡,然后查看是否支持GPU运算。
如果支持GPU运算就可以安装CUDA
下载地址如下:
cuda8下载地址
cudnn下载地址
tensorflow1.3需要cuda8+cudann8+v6.0+GPU版tensorslow
下载cuda8之后安装,解压cudann8把cudann8中的三个文件
bin,include,lib下面的文件不是文件夹拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应的文件夹下面。
配置环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64;

然后在控制面板更改使用有GPU的显卡,重启电脑。
测试代码

#-*- coding:utf8 -*-
# python
import os
import te
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值