1. TensorFlow
TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,表达了高层次的机器学习计算,大幅简化了第一代系统,并且具备更好的灵活性和可延展性。
TensorFlow一大亮点是支持异构设备分布式计算,它能够在各个平台上自动运行模型,从电话、单个CPU / GPU到成百上千GPU卡组成的分布式系统。
TensorFlow支持CNN、RNN和LSTM算法,这都是目前在Image,Speech和NLP最流行的深度神经网络模型。
2015年11月5日,Google开源了TensorFlow,按他们自己的说法,tensorflow会“成为机器学习界的安卓”。所以无论如何,这也表达了TensorFlow强大的性能以及应用的广泛性。无论如何,TensorFlow的开源对于学术界以及工业界,都是非常有价值的。
今天,我们就尝试在Ubuntu16.04+GTX1080上配置和使用这个强大的深度学习工具:TensorFlow。
2. 配置
Ubuntu16.04
GTX1080
CUDA8.0
cudnn5.1
TensorFlow
3. 安装
由于电脑的显卡以及CUDA8.0已经配置好,在此不再赘述,可以参考本人的第一篇文章。
3.1 cudnn
TensorFlow需要用到cudnn,cudnn是NVIDIA开发的用于深度神经网络的GPU加速库。下载地址
这里我们下载的是:Download cuDNN v5.1 (August 10, 2016), for CUDA 8.0,因为我们的显卡是GTX1080以及CUDA8.0,你也可以根据你的显卡来选择你需要的版本。
下载完后解压:
cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz
tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
以下的操作需要管理员权限。
复制头文件:
cd cuda/include
sudo cp cudnn.h /usr/local/cuda-8.0/include
再将lib64目录下的动态文件进行复制和链接:
cp lib* /usr/local/