TensorFlow的安装及使用
一、 TensorFlow的基本介绍
TensorFlow可被用于语音识别或图像识别等多项机器学习和深度学习领域,对2011年开发的深度学习基础架构DistBelief进行了各方面的改进,它可在小到一部智能手机、大到数千台数据中心服务器的各种设备上运行。TensorFlow将完全开源,任何人都可以用。
其既可以在纯CPU上运行,也可以在携有英伟达(NVIDA)GPU的机器上进行网络训练(AMD显卡也有第三方.whl包),一般来讲,GPU版本下TensorFlow性能更强。
TensorFlow由C++编写,在绝大部分应用场景下,开发者使用Python进行调用及开发。
二、 Windows下安装TensorFlow
截止本文书写,Windows平台上仅支持Python 3.5.x(64位)。
首先安装Python,网址为https://www.python.org/downloads/windows/。可以选择3.5.3版本下载并安装,如果电脑上没有其他版本的Python,则可以在安装时选择“添加至Path”;若已安装其他版本的Python,则建议手动管理Path以避免冲突。
如果需要安装GPU版本的tensorflow,那么需要安装CUDA。建议下载9.0版本,下载地址为https://developer.nvidia.com/cuda-90-download-archive,不要选择9.1,否则最后会喝着酒并深刻地反思自己的所作所为。
接下来还需要安装CUDNN,访问https://developer.nvidia.com/rdp/cudnn-download网站(该网站需要注册),选择合适的版本进行下载,CUDNN版本号需要是7,对应CUDA版本为9.0。下载之后解压缩,将.h,.lib和.dll 文件分别拷贝到CUDA的include, lib/x64, bin 文件夹下。根据经验,CUDA大致路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA。
安装结束以后,更新pip:python -m pip install --upgrade pip。紧接着在PowerShell(或者CMD)中输入:python -m pip install tensorflow以及python -m pip install tensorflow-gpu分别进行CPU或GPU版本的安装。
全部安装完毕以后,打开Python Shell,输入“import tensorflow as tf”,如果没有报错,则应该是装好了。
三、 Linux下安装TensorFlow
本次安装在Deepin Linux上完成。
同样,首先安装Python。CPU版本的TensorFlow直接使用pip安装,而且Python2或是Python3均可:python -m pip install tensorflow。
若是需要安装GPU版本的TensorFlow,首先需要安装独显驱动,既可以安装闭源的官方驱动,也可以安装开源的大黄蜂方案(对于Deepin,建议使用大黄蜂方案,对于Ubuntu,可以尝试闭源驱动)。
接着安装CUDA,Linux下使用sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler安装CUDA相关依赖。
CUDNN的版本需要和CUDA一致,可以到官网上进行下载(网址:https://developer.nvidia.com/cudnn)。解压之后,将cuda/include/下的cudnn.h文件复制到/usr/local/include/目录下,然后将cuda/lib64/目录下的所有文件复制到/usr/lib/目录下。
最后,输入pip install tf-nightly-gpu,安装完毕。
四、 MacOS下安装TensorFlow
MacOS下安装TensorFlow大体上需要三个步骤:
安装Python以及pip,sudo easy_install pip,sudo easy_install --upgrade six。然后设置TF_BINARY_URL,对于Python2输入:export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-1.1.0-py2-none-any.whl,对于Python3输入:export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-1.1.0-py3-none-any.whl。最后通过pip安装TensorFlow,sudo pip install --upgrade $TF_BINARY_URL。
全部安装完毕以后,打开Python Shell,输入“import tensorflow as tf”,如果没有报错,则应该是装好了。
五、 TensorFlow的简单使用
安装完以后,可以通过一个简单的demo来熟悉TensorFlow的基本使用。推荐博文:
-> https://blog.csdn.net/geyunfei_/article/details/78782804