tensorflow框架学习
文章平均质量分 57
Zjhao666
泰山不择细壤,故能成其大;江河不择细流,故能成其深。
展开
-
torch、tensorflow1 GPU测速代码python3(MNIST,简易版)
1、torchimport torchimport torch.nn as nnimport torch.nn.functional as Fimport numpy as npimport matplotlib.pyplot as pltfrom torch.autograd import Variableimport time# print start timestart = time.time()print("Start time = " + time.ctime())#原创 2021-10-26 09:35:23 · 920 阅读 · 0 评论 -
深度学习训练降低显存分析
这一篇博客主要是帮助分析显存为什么炸裂的,不单独提具体的技术。技术可以参见另一篇博客深度学习训练降低显存指南_Zjhao666的博客-CSDN博客首先,从总体来看,显存的增长情况大致可以分为两类:线性型:以epoch或更短的batch为单位,显存整体上呈线性缓慢增长。 波动型:没有固定的周期,短期内显存急剧增加,然后裂开了。其次,从显存的分配来看,分配对象一般分为两类:模型变量:主要是模型的固定参数(显存固定),以及一些如结构形的增量参数(显存渐增) 数据变量:包括模型的输入输出,原创 2021-08-19 20:31:28 · 592 阅读 · 0 评论 -
ubuntu18.04利用docker安装NVIDIA驱动、cuda、cudnn、tensorflow、torch等深度学习环境
目录前言一、Docker安装备份sources.list并新建阿里源镜像填入(最近清华源好像有问题)更新源使之生效二、NVIDIA GPU 驱动安装驱动安装检查安装结果三、Docker安装docker安装检查安装结果为避免docker操作必须要sudo,将当前用户加入docker用户组(${YOUR_NAME_HERE}处填当前用户名)前言docker一词的原意为“码头的装卸工”,其开发的目的是为解决广大程序员在开发、测试、部署维护等不同场景不可避免的原创 2020-06-06 17:16:25 · 5790 阅读 · 0 评论 -
ubuntu下同时安装ipython2和ipython3
这个我在我机子上试过没有问题.ipython3安装在ubuntu上应该都问题不大,因为有python3的支持,但是ipython2的安装多少都有些问题因为ipython高版本不支持python2.7,而ipython2若安装不当又会和ipython3冲突....经过一番查阅之后,我找到一个能够同时安装ipython2和ipython3的方式:先安装ipython3,再安装ipython...原创 2019-03-04 14:14:51 · 1332 阅读 · 0 评论 -
win10系统人工智能框架安装指南及常见问题解决方案——anaconda、python27、python35、numpy、tensorflow-gpu、matplotlib、caffe及中间的相关插曲
前言本来这篇文章是可以不用写的,如果NVIDIA给linux设计了RTX20系列notebook驱动的话,但可惜没有(现在有了,233),所以新买的笔记本无法使用caffe和TensorFlow的gpu版本。如果用cpu版本,i7还比不上MX150显卡,所以,有必要研究一下如何在现有的win10系统上安装AI的相关框架和驱动等配套程序(事实证明这也折腾了两三天,主要是caffe折腾的,这货在...原创 2019-03-07 21:33:28 · 2422 阅读 · 0 评论 -
TensorFlow 全连接层 卷积层 池化层 单向lstm 双向lstm 的基本操作
首先导包import tensorflow as tfimport tensorflow.contrib as contribimport tensorflow.layers as layerimport tensorflow.nn as nn全连接层batch = 8data_dim = 10out_dim = 15data = tf.Variable(tf.rand...原创 2019-10-07 15:24:20 · 2332 阅读 · 0 评论 -
TensorFlow分batch加载并生成数据
import tensorflow as tfimport pandas as pdimport timesess = tf.InteractiveSession()def gen(): csv_data = pd.read_csv('../dataset/train.csv') length = csv_data.shape[0] for i in ran...原创 2019-10-07 15:32:50 · 485 阅读 · 0 评论