您的位置:首页 > 其它

Windows 7(10) 下安装 tensorflow,theano,keras并使用GPU加速。

2017-04-07 18:57 603 查看
1.python安装(有python3.5版本的可以跳过此步)
直接下载安装anaconda,选择对应python 3.5版本(因为windows的tensorflow不支持python 2.7),我的是anaconda4.1.1 对应python 3.5.2
下载地址:
https://repo.continuum.io/archive/index.html

2.tensorflow安装
命令行,使用pip安装:
>pip install tensorflow-gpu

cpu版本:
>pip install tensorflow

提示pip版本较低或找不到资源,升级pip:
>python.exe -m pip install --upgrade pip

安装好后测试,命令行输入:
>python
>>>import tensorflow as tf

3.使用GPU加速
下载cuda:https://developer.nvidia.com/cuda-downloads
选择cuda8.0,windows对应版本
直接安装即可
下载cudnn:https://developer.nvidia.com/cudnn 选择5.1版本
下载后解压lib等三个文件夹替换到cuda安装目录中。
这时候tensorflow应该可以使用GPU了,测试使用gpu加速:http://blog.csdn.net/freedom098/article/details/51918825(测试代码,使用cnn对mnist分类)
4.keras安装
pip install keras,会带上theano一起安装,windows下默认以tensorflow为底层。
keras测试代码:http://www.cnblogs.com/Mu001999/p/6221096.html
keras切换底层:http://blog.csdn.net/tangwenbo124/article/details/52627774?locationNum=7
 

5.theano配置GPU

直接在当前用户目录下(如:C:\Users\Administrator)新建一个.theanorc.txt文件,内容:
[global]
floatX = float32
device = gpu
[nvcc]
flags=-LD:\Anaconda\libs
compiler_bindir=C:/Program Files (x86)/Microsoft Visual Studio 10.0/VC/bin
把nvcc中的内容
换成自己的Anaconda\libs目录和vs的目录
         命令行下,测试: 

>python
>>>import theano
>>>theano.test()
         

        
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息