tensorflow
ACTerminate
这个作者很懒,什么都没留下…
展开
-
tf.nn.embedding_lookup函数的用法
tf.nn.embedding_lookup(params,ids)是用于查找在params中取出下标为ids的各个项import numpy as npimport tensorflow as tfx = 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 · 296 阅读 · 0 评论 -
tensorflow-gpu版本相对应的CUDA版本
Linux**tensorflow 版本 **Python 版本编译器编译工具cuDNNCUDAtensorflow_gpu-1.13.12.7、3.3-3.6GCC 4.8Bazel 0.19.27.410.0tensorflow_gpu-1.12.02.7、3.3-3.6GCC 4.8Bazel 0.15.079tensorflow...原创 2019-07-31 21:14:06 · 6690 阅读 · 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 --versioncat /usr/local/c...原创 2019-07-31 21:36:05 · 1440 阅读 · 1 评论