Anaconda安装tensorflow-gpu小记
1 安装Anaconda
可以参考这篇文章链接: windows下配置安装Anaconda+tensorflow.基本上一路next就完事了,比较简单。
总之记住版本首选稳定版,组件之间版本号必须契合才能work。
2 安装tensorflow-gpu
如果你的设备有GPU硬件,那么推荐直接安装tensorflow-gpu版本跳过cpu版本。
2.1 确定tf组件之间的版本
点击下方的链接去tensorflow官网找到你的python适合的tf版本、cuda版本以及cuDNN版本(后两个是nvidia公司为tf特别推出的驱动程序和硬件加速套件)。
TF中文官方网站.
往下翻找到从源代码构建,有一个表格展示了tf、python、编译器、构建工具、cuDNN以及CUDA组件之间的搭配关系。
然后从下面两个链接去下载CUDA和cuDNN。
CUNA下载链接(PS:可以下network版本)
cuDNN下载链接(PS:下这玩意的时候费了点时间