Teachfeeling2.5.0怎么安装
发布网友
发布时间:2022-04-29 22:35
我来回答
共1个回答
热心网友
时间:2022-06-24 17:20
1、创建虚拟环境
通过该链接在 Windows 环境中从源代码构建 | TensorFlow (google.cn)
tensorflow2.5和python=3.7的版本对应。因此创建虚拟环境如下:
conda create --name TF2.5 python=3.7
conda env list #查看是否创建成功
2、激活环境
activate TF2.5
3、安装tensorflow GPU2.5
pip install tensorflow-gpu==2.5.0 -i https://pypi.tuna.tsinghua.e.cn/simple
4、在tensorflow2.x版本下,使用如下代码进行测试:
import tensorflow as tf
tf.compat.v1.disable_eager_execution()
sess = tf.compat.v1.Session()
a=tf.constant(1)
b=tf.constant(2)
print(sess.run(a+b))