
tensorflow
ACTerminate
这个作者很懒,什么都没留下…
展开
-
tf.nn.embedding_lookup函数的用法
tf.nn.embedding_lookup(params,ids)是用于查找在params中取出下标为ids的各个项 import numpy as np import tensorflow as tf x = np.array([[1,2,3],[4,5,6],[7,8,9]]) x = tf.convert_to_tensor(x) a = tf.nn.embedding_lookup(x...原创 2019-04-07 16:17:23 · 323 阅读 · 0 评论 -
tensorflow-gpu版本相对应的CUDA版本
Linux **tensorflow 版本 ** Python 版本 编译器 编译工具 cuDNN CUDA tensorflow_gpu-1.13.1 2.7、3.3-3.6 GCC 4.8 Bazel 0.19.2 7.4 10.0 tensorflow_gpu-1.12.0 2.7、3.3-3.6 GCC 4.8 Bazel 0.15.0 7 9 tensorflow...原创 2019-07-31 21:14:06 · 6797 阅读 · 0 评论 -
cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
在我实验中使用tensorflow-gpu版本的时候,跳出了CUDA driver version is insufficient for CUDA runtime version. 问题就是CUDA版本与tensorflow版本不对应,首先就是要确保安装了相应的正确的tensorflow版本 查看cuda版本 #linux下两种方法 nvcc --version cat /usr/local/c...原创 2019-07-31 21:36:05 · 1502 阅读 · 1 评论