当前位置:主页 > 软件编程 > Python代码 >

查看已安装tensorflow版本的方法示例

时间:2020-12-16 09:31:26 | 栏目:Python代码 | 点击:

由于tensorflow版本不同,可能一些函数的调用也有变换,这时候可能需要查看tensorflow版本,可以在终端输入查询命令如下:

import tensorflow as tf

tf.__version__

查询tensorflow安装路径为:

tf.__path__

查询结果如下:

根据自己的情况选择以下命令之一进行安装:

pip install tensorflow==1.2   # Python 2.7; 仅支持CPU
pip3 install tensorflow==1.2   # Python 3.n; 仅支持CPU 
pip install tensorflow-gpu==1.2   # Python 2.7; 支持CPU
pip3 install tensorflow-gpu==1.2   # Python 3.n; 支持CPU

注意这里的==1.2.是表示tensorflow的版本,请根据实际情况修改。cuda8.0对应的是1.2 。

cuda与tensorflow的版本一定要对应,不然会出错!-------重要

下图是对照表:

 验证Tensorflow是否安装成功

启动终端,输入python,如下:

python
>>> import tensorflow as tf 
>>> hello = tf.constant('Hello, TensorFlow!') 
>>> sess = tf.Session() 
>>> print(sess.run(hello))

如果输出Hello, TensorFlow!则代表安装成功,如下图。

Tensorflow卸载:

sudo pip uninstall protobuf
sudo pip uninstall tensorflow

您可能感兴趣的文章:

相关文章