如何安装 TensorFlow (Windows 环境下)

2017-10-14  本文已影响0人  智慧与力量的贤者

前置检查


使用 pip 安装 TensorFlow


安装 cuda

划重点!!!! 在安装之前,请务必关闭 360安全卫士/腾讯管家 等安全软件!!!


下载 cudnn


检查是否能正常使用


补充:

  1. 简单矩阵乘法 测试示例代码:
import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)

输出:

output
上一篇 下一篇

猜你喜欢

热点阅读