码迷,mamicode.com
首页 > 编程语言 > 详细

pycharm+annaconda3+python3.5.2 + 安装tensorflow-gpu版本 [gtx 940mx + Cuda7.0+cudnn v4.0 ]

时间:2017-11-01 01:13:00      阅读:610      评论:0      收藏:0      [点我收藏+]

标签:idt   png   下载   完成   python3   pycharm   under   uda   height   

1、安装cuda Toolkit 和cudnn (百度云可下载,版本需要对应)

技术分享

2、配置环境变量:

技术分享

技术分享

3、安装cudnn(需要拷贝一些dll和lib来进行配置)

技术分享

4、进入cmd,找到anaconda3的pip路径,用下面的命令来执行,可以卸载cpu版本的tensorflow,安装gpu版本的tensorflow

pip uninstall tensorflow
pip install tensorflow-gpu

完成,tensorflow在训练时候会自动调用gpu来进行计算

 

pycharm+annaconda3+python3.5.2 + 安装tensorflow-gpu版本 [gtx 940mx + Cuda7.0+cudnn v4.0 ]

标签:idt   png   下载   完成   python3   pycharm   under   uda   height   

原文地址:http://www.cnblogs.com/jerrybaby/p/7764560.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!