【tensorflow学习之路】如何使用gpu进行运算

准备阶段：

Anaconda 3（官网下载）
python3.x（Anaconda3自带）
GPU（最低NVDIA GTX 650）
tensorflow-gpu（Anaconda中安装）
tensorflow基本语法操作

代码测试

写一个手动调用gpu设备的代码

# coding:utf-8
'''
**************************************************
@File   ：深度学习实战[安东尼奥] -> Cap1-hello world
@IDE    ：PyCharm
@Author ：Small_wind
@Date   ：2019/11/17 10:46
**************************************************
'''
import tensorflow as tf##1.先写个hello world测试下
message=tf.constant("hello world")#常量字符串sess=tf.Session()
print(sess.run(message))
sess.close()##2.常量、变量
#略过##3.调用GPU/CPU设备
c=[]
with tf.device('/gpu:0'):rand_t=tf.random_uniform([50,50],0,10,dtype=tf.float32,seed=0)a=tf.Variable(rand_t)b=tf.Variable(rand_t)c.append(tf.matmul(a,b))init=tf.global_variables_initializer()#要验证Tensorflow是否确实再使用指定设备，可设置log_device_placement=True
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))
sess.run(init)
print(sess.run(c))
sess.close()

结果

当然，在Anaconda安装了tensorflow-gpu版本后，之后运行程序时，会自动优先调用gpu的（如果有gpu）

【tensorflow学习之路】如何使用gpu进行运算相关推荐

我的Tensorflow学习之路
最近两年深度学习真的是火的不要不要的,关于深度学习,每个人都有自己的看法.有人说就是炼丹,得个准确率召回率什么的,拿到实际中,问问为什么,都答不上来.各种连代码都没写过的人,也纷纷表示这东西就是小孩堆 ...
【TensorFlow学习之路】一--TensorFlow基本概念
计算图计算图是TensorFlow中最基本的一个概念,TensorFlow中的所有计算都会被转化为计算图上的节点. TensorFlow 的名字中己经说明了它最重要的两个概念一一Tensor 和 F ...
[ Tensorflow学习之路 ] —— API：TF-Slim
https://www.hongweipeng.com/index.php/archives/1297/ https://github.com/tensorflow/tensorflow/tree/m ...
《30天吃掉那只 TensorFlow2.0》开篇辞（Tensorflow 学习之路）
<30天吃掉那只 TensorFlow2.0 >开篇辞
菜鸟de深度学习之路——（2）张量运算和梯度优化
一,引言上一节https://blog.csdn.net/zzl1060549268/article/details/88675915通过一个具体的例子,从整体上鸟瞰了一下一个三层神经网络,但是对于 ...
Tensorflow学习笔记6：解决tensorflow训练过程中GPU未调用问题
Tensorflow学习笔记6:解决tensorflow训练过程中GPU未调用问题参考文章: (1)Tensorflow学习笔记6:解决tensorflow训练过程中GPU未调用问题 (2)http ...
检验 pytorch,tensorflow，paddle，mxnet 深度学习框架是否正确支持GPU功能
检验 pytorch,tensorflow,paddle,mxnet 深度学习框架是否正确支持GPU功能 1.pytorch 框架 import torch a = torch.cuda.is_ava ...
tensorflow学习笔记：查看tensorflow可配置运算资源以及配置使用GPU运算
查看tensorflow可配置运算资源以及配置使用GPU运算因为还用不到分布式的tensorflow,自己没有尝试过所以就不写分布式tensorflow的使用了(等自己用上了再说),这里记录一下在跑 ...
tensorflow学习入门笔记
<div class="note"><div class="post"><div class="article" ...

【tensorflow学习之路】如何使用gpu进行运算

准备阶段：

代码测试

结果

【tensorflow学习之路】如何使用gpu进行运算相关推荐

最新文章

热门文章