以下为安装教程,环境配置请拉倒文尾,另外注意安装顺序,一定要你先安装caffe-gpu,再安装tensorflow-gpu,因为我的教程安装tensorflow-gpu过程中会用到caffe-gpu安装过的cuda8.0、cudnn5.1等
第一步 安装caffe-gpu
第二步 安装tensorflow-gpu
利用前面安装好的cuda cudnn环境
ubuntu下tensorflow-gpu环境搭建(通过anaconda,手动安装或环境含有cuda和cudnn)
环境
- ubuntu64位
- anaconda, conda 4.3.30
- cuda8.0
- cudnn5.1
- GTX970M
- nvidia driver 384
参考资料
1.文章内链接