使用GPU加速TensorFlow机器学习

最新推荐文章于 2024-07-26 10:42:33 发布
最新推荐文章于 2024-07-26 10:42:33 发布
阅读量6.1k 收藏 5
点赞数
分类专栏： 实用技术 文章标签： tensorflow gpu cnn 机器学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/gongxsh00/article/details/79839149
版权

    在Thinkpad X260上运行一个CNN图像分类的样例程序时，发现速度特别慢，迭代一轮要将近5分钟，那么迭代200轮需要1000分钟，16个小时！在看到TensorFlow相关的书籍时，总是提到GPU加速，对于这样的问题，应该能派上用场吧。TensorFlow目前只支持NVIDIA显卡，笔者手头没有，阿里云上有GPU计算型ECS，便“按量付费”买了一台，开始了一场“坎坷小贵”的验证之旅。坎坷的原因是没有找到一篇完整的指南，遇到各种的版本问题，Tensorflow作为咕果的产品官网打不开；小贵是因为服务器12元/小时，后来看到NVIDIA Tesla P100一块卖40000元，好吧，也不是阿里云抢咱的钱。下面一起来如何在阿里云GPU计算型ECS服务器上运行TensorFlow机器学习。

友情提示：本文中使用的版本已经经过验证，直接使用这些版本可以避免很多“坑”

购买阿里云GPU计算型ECS

    笔者购买的是ecs.gn5-c4g1.xlarge实例规格，使用Ubuntu 16.04 64位操作系统


安装NVIDIA CUDA Toolkit

    NVIDA CUDA Toolkit中含有显卡驱动，不要提前安装显卡驱动，以免冲突。在NVIDA网站上找到

最低0.47元/天 解锁文章
gongxsh00
关注
专栏目录
使用GPU 训练Tensorflow/Keras 的CNN模型
出门搔白首，若负平生志
06-03 2万+
关键词：Kaggle 猫狗大赛，MX510 GPU， 联想潮7000， Win10， NVIDA显卡之前写了一个猫狗识别的CNN模型，利用笔记本进行训练，每次都需要好久，基本每个epoch要5分钟左右，来来回回改改参数，每次都要等漫长的时间。于是在找怎么利用GPU进行训练。1. 电脑的显卡是NVIDA MX510， 可以支持CUDA，要使用tensorflow-gpu版本，需要安装CUDA9.0，...
Deep Learning-TensorFlow (10) CNN卷积神经网络_ TFLearn 快速搭建深度学习模型
DiamonJoy
04-10 7199
TFLearn 是一个构建在 TensorFlow 之上的模块化和透明的深度学习库。它为 TensorFlow 提供高层次 API，目的是便于快速搭建试验环境，同时保持对 TensorFlow 的完全透明和兼容性。
Pytorch机器学习代码GPU使用示范
weixin_47390487的博客
02-08 687
自己买了带GPU机器但长期只能狂跑CPU，最近通过借鉴代码和摸索，终于让自己的GPU跑起来了。为了更方便的说明，我把自己参加的一个天池长期赛部分代码展示出来，并突出GPU相关的修改部分。
GPU介绍和入门知识整理
最新发布
Chenzhinan1219的博客
07-26 2506
GPU入门介绍知识
Tensorflow使用GPU加速
weixin_30867015的博客
08-13 114
测试faster-rcnn时，cpu计算速度较慢，调整代码改为gpu加速运算 将with tf.Session() as sess:替换为 1 gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.9) 2 with tf.Session(config=tf.ConfigProto(gpu_op...
猫狗识别基于tensorflow2.0 GPU版 自建CNN模型
wmoxie的博客
06-01 796
基于tensorflow2.0 GPU版 自建CNN模型 1. 导入库 import tensorflow as tf from tensorflow.keras.layers import Conv2D, MaxPooling2D, Flatten, Dense from tensorflow.keras.models import Sequential, load_model from tensorflow.keras import optimizers from tensorflow.keras.pr
TensorFlow开源机器学习框架机器学习模型
06-09
- **GPU加速**：通过CUDA和cuDNN等技术，TensorFlow能够在GPU上执行计算密集型任务，显著提升训练速度。 - **并行计算能力**：支持数据并行和模型并行两种方式，可以利用多核处理器或多GPU来提高计算性能。 3. **...
谷歌TensorFlow机器学习框架及应用.pptx
10-20
谷歌TensorFlow机器学习框架及应用 一、TensorFlow机器学习框架简介 TensorFlow是谷歌开发的一款开源机器学习框架，它采用数据流图（Data Flow Graph）的方式描述计算过程，能够高效地处理大规模数据。...
TensorFlow机器学习、深度学习.zip
05-08
这个名为"TensorFlow机器学习、深度学习.zip"的压缩包可能包含了关于使用TensorFlow进行机器学习和深度学习的教程、代码示例、笔记或者项目。让我们深入探讨一下TensorFlow及其在这些领域的应用。 首先，TensorFlow...
在tensorflow中设置使用某一块GPU、多GPU、CPU的操作
09-18
随着深度学习的发展，TensorFlow作为一款强大的机器学习框架，为开发者提供了灵活且高效的编程接口。为了更好地利用硬件资源，特别是在多GPU环境下的训练加速，正确地配置运行设备变得尤为重要。本文将详细介绍如何...
tensorflow2.5.0-GPU测试工程
07-27
在ubuntu16.04上编译好tensorflow2.5.0-GPU；使用的是RTX 2060 Super测试多张图片；比TF-CPU加速很多。
tensorflow下编写CNN网络的框架
06-08
Windows下tensorflow-GPU-1.8的python下的CNN模板，内置的lenet-5模型，我特意把它修改成很容易换成别的网络模型，我是在官网的cifar-10代码的基础上进行改动，里面写了详细的中文注释，我还加上了获取混淆矩阵和分类错误图片的路径的功能，更加方便分析模型性能。
使用GPU训练Tensorflow/Keras 的CNN模型
yiyayiya557的博客
05-07 1471
参考文章： 使用GPU 训练Tensorflow/Keras 的CNN模型 https://blog.csdn.net/xiaozisheng2008_/article/details/80558914 tensorflow显存管理、tensorflow使用多个gpu训练 https://blog.csdn.net/liulina603/article/details/80180355 ...
Faster RCNN - Tensorflow实现以及训练自己的数据 — GPU版（排坑）
Pngyng的博客
04-29 3018
转载请注明： 参考： GPU坑：https://blog.csdn.net/hongxingabc/article/details/79048531 GPU： 1.Environment 创建环境 Centos7 CUDA：0.9 CUDNN：7.3.1 TeslaP100：sm_60 conda create --name tensorflow-gpu python=3.6 conda inst...
关于GPU下tensorflow加速CNN的运行操作步骤
qq_31648921的博客
12-01 1014
1，已按照https://blog.csdn.net/qq_31648921/article/details/96426915 安装好GPU加速相关组件 2，创建好了envs 3,具体的操作步骤：
gpu填充速率 计算_GPU如何加速人工智能或机器学习的计算速度
weixin_29331115的博客
12-24 304
一、Why GPU其实GPU计算比CPU并不是“效果好”，而是“速度快”。计算就是计算，数学上都是一样的，1+1用什么算都是2，CPU算神经网络也是可以的，算出来的神经网络放到实际应用中效果也很好，只不过速度会很慢罢了。GPU的起源GPU全称叫做graphics processing unit，图形处理器，顾名思义就是处理图形的。电脑显示器上显示的图像，在显示在显示器上之前，要经过一些列处理，这个...
【Tensorflow】图解tf.image.extract_patches的用法--提取图片特定区域
BetterBench的博客
06-08 4029
目录1 引言2 图解2.1 Demo12.2 Demo2 1 引言 （1）作用 提取图片中的一小块区域，也可以用于分割图片为多个子图片。类似卷积核提取特征，但不一样 （2）用法 tf.image.extract_patches( images, sizes, strides, rates, padding, name=None ) 参数： images：图像矩阵，形状必须是[batch, in_rows, in_cols, depth] sizes：pat
求助！ tensorflow 中，CNN 在GPU上运行无响应，然后内核崩溃
weixin_41014532的博客
04-01 727
CNN在GPU上运行卡住，然后内核挂掉，但是在CPU上能跑。尝试了多种办法，均没有效果，向大佬求救！！
Ubuntu环境下搭建GPU加速TensorFlow开发环境教程
TensorFlow是一个强大的开源机器学习框架，通过GPU加速可以显著提高计算速度。" 在构建这个环境时，首先需要下载CUDA 8.0。CUDA的下载地址在NVIDIA的开发者网站上，根据GPU型号（例如GTX 1050、GTX 1070、GTX 1080...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值