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

2020-11-05 10:17:32 浏览数 (1)

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

代码语言:javascript复制
import tensorflow as tf

tf.__version__

查询tensorflow安装路径为:

tf.__path__

查询结果如下:

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

代码语言:javascript复制
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,如下:

代码语言:javascript复制
python
    import tensorflow as tf 
    hello = tf.constant('Hello, TensorFlow!') 
    sess = tf.Session() 
    print(sess.run(hello))

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

Tensorflow卸载:

代码语言:javascript复制
sudo pip uninstall protobuf
sudo pip uninstall tensorflow

到此这篇关于查看已安装tensorflow版本的方法示例的文章就介绍到这了,更多相关查看已安装tensorflow版本内容请搜索ZaLou.Cn

0 人点赞