torch、tensorflow-cpu

2019-03-07  本文已影响0人  青草_e75f

import torch

print(torch.__version__)

print(torch)

torch版本
torch安装路径

pyana37

import tensorflow as tf

tf.__version__

tf.__path__

tensorflow-cpu版本

查看py35下安装的tensorflow版本
import numpy import tensorflow as tf

a = tf.constant([1.0,2.0,3.0,4.0,5.0,6.0], shape=[2,3], name='a')

b = tf.constant([1.0,2.0,3.0,4.0,5.0,6.0], shape=[3,2], name='b')

c = tf.matmul(a, b)

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) print(sess.run(c))

tensorflow-Gpu
上一篇 下一篇

猜你喜欢

热点阅读