【发布时间】：2016-12-13 07:14:55
【问题描述】：
我尝试运行 tf.one_hot，得到 CUDA_ERROR_LAUNCH_FAILED 错误。详情如下：
示例代码：
import tensorflow as tf
idx_0 = tf.placeholder(tf.int64, [None])
mask = tf.one_hot(idx_0, 3, axis=-1)
sess = tf.Session()
sess.run(tf.global_variables_initializer())
a = sess.run([mask],feed_dict={idx_0:[0,1,2]})
print(a)
预期结果：
[array([[ 1., 0., 0.],
[ 0., 1., 0.],
[ 0., 0., 1.]], dtype=float32)]
实际结果：
E c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\cuda\cuda_driver.cc:1177] could not synchronize on CUDA context: CUDA_ERROR_LAUNCH_FAILED :: No stack trace available
E c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\stream_executor\cuda\cuda_event.cc:49] Error polling for event status: failed to query event: CUDA_ERROR_LAUNCH_FAILED
F c:\tf_jenkins\home\workspace\release-win\device\gpu\os\windows\tensorflow\core\common_runtime\gpu\gpu_util.cc:370] GPU sync failed
电脑配置：
- TensorFlow 0.12.0-rc1
- Python 3.5
- CUDA 8.0
- cuDNN 5.1
- 操作系统：Windows 10
- GPU：GeForce GTX 970
tf.one_hot 在 Linux CPU、Linux GPU (GeForce GTX 660)、Windows 10 CPU 上运行正常。在 Windows 10 GPU 上不行。
在 Windows 10 GPU 上，tf.matmul、tf.reduce_mean、tf.reduce_sum 运行正常。但是 tf.one_hot 不行。
这是一个错误，还是我错过了什么？谢谢。
（编辑 2016-12-16）
我已经在 GPU 的 Xubuntu 中的同一台机器上运行了代码。代码运行良好。所以我认为这是 TensorFlow-Windows 中的一个问题。
【问题讨论】：
标签： tensorflow