tensorflow-gpu环境的两种安装方式

来源:互联网 发布:python 量化 书籍 编辑:程序博客网 时间:2024/06/18 03:49

Windows安装建议使用win10,cuda和cudnn对win10支持比较好!
1. 使用winPython来制造,直接上winpython的官网下载并安装,安装完毕之后,可以到安装目录下启动控制台,这个时候可以在这样的控制台上安装各种Python-package,至于想要什么样的package可以到pipy的主页上去寻找和download,当然我们下载应该是tensorflow-gpu版本,以及再下载numpy,至于版本要求可以在主页上面看要求!安装扩展包指令pip install 包路径
2. 另外一种方式是安装anaconda,在anaconda下启动控制台,扩展包同上,只是安装命令有点不同,使用的是conda install 包路径
3. 上面是两种方式都是用于制造TensorFlow-gpu的环境,接下来还需要到nvidia,也就NVIDIA的gpu支持TensorFlow,到cuda上下载8.0工具,选择win10版的(根据你系统,当然建议你的系统是win10),再下载cudnn文件(根据系统和cuda的版本下载所需要的文件),将cudnn里面的include文件和bin,lib文件都拷贝到cuda安装目录里面的对应的文件下里面,再系统的环境变量添加cuda的bin路径和lib路径。
4. 一切准备好了,上面都是环境,接下来就是开发工作,这个时候需要ide,建议使用pycharm,自行下载,然后将我们制造的环境import到pycharm就可以开始愉快的玩耍了,pycharm仅仅是个ide没办法编译库,因此就需要import!!
5. 另外一点说明,安装gpu版本是可以兼容CPU的,也是可以指定代码到cpu上跑一跑!

原创粉丝点击