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
网友评论