tensorflow是谷歌所开发的用于深度学习,并完全开源的一个库,我也想通过学习tensorflow来逐渐了解深度学习以及机器学习。和以往学过的内容一样,第一步要做的就是配环境。
tensorflow版本详情
tensorflow在今年支持windows上的安装使用了,如今最新为1.3版本,可能还不是很通用,更多使用的是1.1和1.2版本,根据官网所说,仅支持python3.5版本,不过我使用的python3.6也安装成功了,应该是3.5以后版本都能用。在windows上使用tensorflow,还分为CPU和GPU两版,GPU需要NVIDIA显卡支持,同时安装也更麻烦,不够考虑到CPU的运算速度,还是决定安装GPU版本。
安装所需环境
为成功安装tensorflow的GPU版本,一共需要在电脑安装:
安装(踩坑)详情
以上所需软件也可在官网下载安装,需要注意的有,python安装时需要添加PATH变量以便使用pip快速导入包。然后可使用pip指令快速安装tensorflow库pip3 install --upgrade tensorflow-gpu
虽然不知道为什么,我一开始下载的时候一直报错,可能是网络不稳定吧,总之最后还是能安装的。
VS2015在线安装可能会失败,最好先翻墙。
主要的困难在于CUDA和cuDNN,CUDA安装完后可在命令行输入指令查看nvcc -V
cuDNN下载后需要将文件添加入PATH环境变量,也可将解压后文件夹中的三个子文件夹中的文件分别复制添加至cuda所安装位置下的文件夹,默认安装路径是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,其中对应的bin,include,x64文件夹中。
测试代码
安装完成后可在cmd指令界面打开python,输入以下代码进行测试:
能跑就基本是安装成功了