因为工作需要,不能把电脑系统换成Linux。一直是在虚拟机中安装Ubuntu来运行,后来用到机器学习,到深度学后,虚拟机连从大文件读取数据都out of memory。所以逼的没办法,在windows10下安装Anacode3,在虚拟环境中安装深度学习相关内容。看到网上介绍,说GPU速度是cpu的上百倍,一时心动安装tensorflow gpu版。按照网上的攻略试了半天也没解决问题。绕的我头晕目眩的。最后找着问题的原因是在tensorflow官网上。特写博文一篇,为了让大家少走弯路,也提醒大家,遇到问题第一时间要找官网解决起来事半功倍。
tensorflow官网
https://www.tensorflow.org/install/source#common_installation_problems
关于安装条件说明
各版本配套
大家照着版本对应安装就好。安装前一定是先装anaconda,再装cuda,然后是cudnn,最后是tensorflow-gpu,安装时一定要指定版本。
下面留下cuda和cudnn地址,以及tensorflow-gpu安装方法
cuda 8:
https://developer.nvidia.com/cuda-80-ga2-download-archive
cudnn 5.1:
https://developer.download.nvidia.com/compute/machine-learning/cudnn/secure/v5.1/prod_20161129/8.0/cudnn-8.0-windows10-x64-v5.1.zip?hHIORKW2LYAsR9HBDcIjtSXjxz8KAYyYBaICzLv0vw7DpAuBL0DuoDZ-YzS6dVSoDF5KtCRE_-OgsoUrXDv6Ao8-1mihVGsBMQBhCnF9oXbFrHzmQsmULY2ljXow9IBJ3OTZfWm1aOY806BnXRldP7_DVoMzU58mJ49MQO7D_kzgc9g8o3zA4nAT2zQqbP1zpjaSVg5NcQZAWeI8
tensorflow-gpu安装
pip install tensorflow-gpu==1.2.0
运行速度
我的CPU是i7 6700HQ
显卡是GTX 970
执行一段代码:gpu计算大约是CPU计算的1/10时间