前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >TensorFlow 安装GPU版本

TensorFlow 安装GPU版本

作者头像
Centy Zhao
发布2019-12-26 13:46:47
1.4K0
发布2019-12-26 13:46:47
举报
文章被收录于专栏:icecream小屋

TensorFlow 有两个版本:CPU 版本和 GPU 版本。GPU 版本需要 CUDA 和 cuDNN 的支持,CPU 版本不需要。如果你要安装 GPU 版本,请先确认你的显卡支持 CUDA。我安装的是 GPU 版本,采用 pip 安装方式,所以就以 GPU 安装为例,CPU 版本只不过不需要安装 CUDA 和 cuDNN。

1. 确认你的显卡支持 CUDA

打开网址找到对应的显卡:https://developer.nvidia.com/cuda-gpus

确认显卡是否支持

我这里是 GXT 1060

2.下载并安装 Visual Studio ,因为CUDA需要Visual Studio的支持。

下载地址:https://www.visualstudio.com/zh-hans/

image.png

安装选项:

image.png

3. 下载安装 CUDA Toolkit 9.0

官网网址:https://developer.nvidia.com/cuda-90-download-archive?tdsourcetag=s_pctim_aiomsg

下载:

image.png

安装:

image.png

4. 下载 cuDNN,并解压到目录,把对应的目录地址添加到path环境变量

https://developer.nvidia.com/cudnn

image.png

下载 cuDNN必须需要一个账号,如果没有,注册一个新的账号,登录后才能下载。

下载:

image.png

解压

image.png

添加到环境变量

添加到环境变量

5. 安装 Anaconda,因为这个集成了很多科学计算所必需的库,能够避免很多依赖问题

image.png

把 Anaconda安装目录 添加到环境变量

image.png

6. 使用 pip 安装 tensorflow GPU版本

管理员身份打开cmd

image.png

输入:

pip install --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

-ihttps://pypi.tuna.tsinghua.edu.cn/simple:使用清华的镜像,飞一般的速度。

7. 测试程序:

importtensorflowastfa = tf.random_normal((100,100))b = tf.random_normal((100,500))c = tf.matmul(a, b)#矩阵相乘sess = tf.InteractiveSession()sess.run(c)

执行结果

出现以上结果,说明安装成功了。

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档