时间: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