安装tensorflow(GPU)较tensorflow(CPu)复杂,其中还要额外下载cuda与cuddn。网上类似的教程很多,然而随着更新迭代,很多教程都出现了很多问题,踩了许多坑,给大家分享下。
配置要求:
- python3.5
- cuda8.0
- cuddn V6
- anaconda3
注意
tensorflow(GPU)官方要求的cuda是8.0,cuddn是v6, 现在Nvidia官网上直接下载的话默认是9.2,所以下载cuda时要尤其注意,别下载错了版本。另外tensorflow的版本问题,也踩了不少坑,下面会着重介绍。
步骤
一、下载anaconda3
anaconda是管理python软件包,和python版本切换。许多python库都可以在里面下载,非常方便。
可以直接在官网上下载 https://www.anaconda.com/download/
直接下载python 3.6 version ,之后跟着安装导引走即可。
安装完,可以看到有anaconda prompt,打开来进行指令操作。
二、下载python3.5.x
因为现在这个版本的anaconda默认是python3.6 所以要创建python3.5的环境,通过如下指令进行配置
1. 创建一个名为python3.5的环境,命名为python35