首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

TensorFlow 安装

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的安装。

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20181029G1T2FC00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码

添加站长 进交流群

领取专属 10元无门槛券

私享最新 技术干货

扫码加入开发者社群
领券