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