Ubuntu下安装TensorFlow-gpu思路

来源:互联网 发布:一个淘宝账号值多少钱 编辑:程序博客网 时间:2024/06/05 00:39

1、既然是安装gpu版本,因此显卡驱动必不可少,自行到英伟达官网下载。
2、cuda8.0和cudnn6.0下载,虽然目前都更新到了cuda9.0和cudnn7.0。但对于tensorflow来说,前者支持兼容更佳!安装教程参考http://blog.csdn.net/touch_dream/article/details/78661277
3、安装Python3.5.x版本不要安装最新版的3.6,apt-get install python3.5.x,apt-get install python3-pip, 接着就可以使用pip3安装下载好的TensorFlow-gpu版本以及numpy等。下载Python扩展包链接:https://pypi.python.org/pypi

总结:
主要的难点在于cuda的安装以及cudnn的配置上!请保持耐心,你会享受到成功之后的喜悦!就是这么简单

原创粉丝点击