您的位置:首页 > 其它

Windows10+Anaconda3+Tensorflow gpu版 1.12 +CUDA8 + cudNN5.1完美运行

2019-01-03 16:44 711 查看

因为工作需要,不能把电脑系统换成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时间

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