神舟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))

猜你喜欢

转载自www.cnblogs.com/xbit/p/11300839.html
今日推荐