3.2 Tensorflow
CongyingWang
咸鱼太闲
展开
-
tensorflow学习笔记——0_零碎问题及解决笔记
文章目录1. `tf.argmax`的使用1. tf.argmax的使用tf.argmax(vector, 1):返回的是vector中的最大值的索引号,如果vector是一个向量,那就返回一个值,如果是一个矩阵,那就返回一个向量,这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。示例:import tensorflow as tfimport numpy as np A =...原创 2018-11-21 17:28:15 · 2829 阅读 · 1 评论 -
Linux下查看NVIDIA的GPU使用情况
文章目录查看命令查看命令Nvidia自带了一个nvidia-smi的命令行工具,会显示GPU使用情况:$ nvidia-smi出现如下图:原创 2019-03-21 18:53:26 · 5090 阅读 · 0 评论 -
Linux查看CUDA版本以及cudnn版本号
文章目录1. 查看CUDA版本2. 查看cudnn版本1. 查看CUDA版本方法1: 查看文件cat /usr/local/cuda/version.txt方法2: 命令nvcc --version2. 查看cudnn版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2输出如下:(图中版本...原创 2019-03-21 18:59:12 · 45286 阅读 · 4 评论 -
bazel 出现no such package '@org_tensorflow//tensorflow' 错误
问题在执行 bazel tensorflow serving 的时候出现下面的错误ERROR: error loading package ‘’: in /home/wangcongying/tools/serving/tensorflow_serving/workspace.bzl: Encountered error while reading extension file ‘tens...原创 2019-05-14 18:53:13 · 13194 阅读 · 3 评论 -
tensorflow API——tf.random_uniform 使用
文章目录定义参数示例定义tf.random.uniform( shape, minval=0, maxval=None, dtype=tf.dtypes.float32, seed=None, name=None)参数shape: 1-D整数Tensor或Python数组, 输出张量的形状。minval: 类型的0-D Tensor或...原创 2019-05-22 16:45:11 · 1173 阅读 · 0 评论 -
Linux 中指定使用的GPU
文章目录1 查看GPU 编号及其使用信息2 方法一:在终端执行程序时指定GPU3 方法二:在Python代码中指定GPU1 查看GPU 编号及其使用信息参考文章Linux下查看NVIDIA的GPU使用情况$ nvidia-smi其中左上侧有0、1、2、3的编号,表示GPU的编号,在后面指定GPU时需要使用这个编号。2 方法一:在终端执行程序时指定GPU$ CUDA_VISIBL...原创 2019-05-13 17:19:28 · 25049 阅读 · 1 评论 -
TensorFlow 指定GPU 日志却显示的是第0块
实际上使用的是指定的在tensorflow 中日志打印的是可见GPU的编号如指定 1,2,3 , 日志中是0,1,2指定1 ,日志中的是0,但实际使用的是自己指定的原创 2019-05-13 17:30:21 · 1033 阅读 · 0 评论 -
tensorflow gpu 运行出现OOM错误
文章目录0 问题1 解决1.1 指定GPU1.2 调参1.3 使用CPU0 问题今天跑了一下程序,报了如下的OOM错误ResourceExhaustedError: OOM when allocating tensor with shape[258000,768] and type float on /job:localhost/replica:0/task:0/device:GPU:0 b...原创 2019-05-13 17:42:27 · 6386 阅读 · 0 评论