tensorflow学习笔记

tensorflow学习笔记

  • [1. 安装](#1. 安装)
  • [2. 验证GPU是否可用](#2. 验证GPU是否可用)
  • [3. 静态图说明](#3. 静态图说明)

tensorflow学习过程中记录的笔记

1. 安装

cuda的安装过程略,参考pytorch的那篇笔记。

一个稳定版本的tensorflow2.5版本匹配:

bash 复制代码
conda create -n hi_hand_det python=3.7.0
conda install cudatoolkit=11.3.1
conda install cudnn=8.2.1
pip install tensorflow-gpu==2.5.0

2. 验证GPU是否可用

python 复制代码
import tensorflow as tf
print(tf.test.is_gpu_available())

如果返回为True,说明GPU已经可用。

3. 静态图说明

参考:tensorflow静态图原理

笔记:在模型训练初始化init阶段的最后,需要用任意数值的指定矩阵形状让模型先运行一次,从而生成静态图,这样后续就会直接运行静态图。

相关推荐
井底哇哇18 分钟前
ChatGPT是强人工智能吗?
人工智能·chatgpt
Coovally AI模型快速验证23 分钟前
MMYOLO:打破单一模式限制,多模态目标检测的革命性突破!
人工智能·算法·yolo·目标检测·机器学习·计算机视觉·目标跟踪
AI浩1 小时前
【面试总结】FFN(前馈神经网络)在Transformer模型中先升维再降维的原因
人工智能·深度学习·计算机视觉·transformer
可为测控1 小时前
图像处理基础(4):高斯滤波器详解
人工智能·算法·计算机视觉
ℳ₯㎕ddzོꦿ࿐1 小时前
解决Python 在 Flask 开发模式下定时任务启动两次的问题
开发语言·python·flask
CodeClimb1 小时前
【华为OD-E卷 - 第k个排列 100分(python、java、c++、js、c)】
java·javascript·c++·python·华为od
一水鉴天1 小时前
为AI聊天工具添加一个知识系统 之63 详细设计 之4:AI操作系统 之2 智能合约
开发语言·人工智能·python
Channing Lewis2 小时前
什么是 Flask 的蓝图(Blueprint)
后端·python·flask
倔强的石头1062 小时前
解锁辅助驾驶新境界:基于昇腾 AI 异构计算架构 CANN 的应用探秘
人工智能·架构
B站计算机毕业设计超人2 小时前
计算机毕业设计hadoop+spark股票基金推荐系统 股票基金预测系统 股票基金可视化系统 股票基金数据分析 股票基金大数据 股票基金爬虫
大数据·hadoop·python·spark·课程设计·数据可视化·推荐算法