TensorFlow™是一个基于数据流编程(dataflow programming)的符号数学库,主要用于各类机器学习(machine learning)算法的编程实现,其前身是谷歌神经网络算法库DistBelief[1]。
Tensorflow拥有多层级结构,可部署于各类服务器、PC终端和网页并支持GPU和TPU高性能数值计算,被广泛应用于谷歌内部的产品开发和各领域的科学研究[1-2]。
准备工作
首先需要看一下,你的电脑是否有NVIDIA的显卡,是否支持CUDA。
查看电脑显卡的驱动。
我的电脑是GTX755M,驱动版本375.61 查看驱动版本的方法见这个链接. 根据这里的表,判断一下该下哪个版本的CUDA。
对于本机,应该下载的是CUDA 8.0版本。
cuDNN的安装方法。
下载cuDNN 6.0 解压之后,将对应的文件夹里面的文件,复制到上文中安装的CUDA的对应文件夹下。 默认的CUDA目录: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
验证是否安装成功
全部安装完之后,可以用这个链接里面的DLtftest.py
执行结果如下:
这样就完成了TensorFlow的安装。
领取专属 10元无门槛券
私享最新 技术干货