神舟G7-CT7NK 安装tensorflow-gpu


参考https://www.cnblogs.com/xbit/p/9768238.html

直接安装,运行keras mnist数字识别报错:

Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

运行如下代码OK:

import tensorflow as tf
a = tf.add(tf.constant(1),tf.constant(2))
print(sess.run(a))

相关