如何安装并查看tensorflow-gpu版本及可否使用

如何安装并查看tensorflow-gpu版本及可否使用

首先，使用以下语句安装tensorflow：

conda install tensorflow-gpu==你要安装的TensorFlow版本

如果要安装的是cpu版本的tensorflow，则在安装时去掉_gpu即可，即：

conda install tensorflow==你要安装的TensorFlow版本

然后输入python之后进行验证，具体代码：

import tensorflow as tf
version=tf.__version__  #输出tensorflow版本
gpu_ok=tf.test.is_gpu_available()  #输出gpu可否使用（True/False）
print("tf version:",version,"\nuse GPU:",gpu_ok)
tf.test.is_built_with_cuda()  # 判断CUDA是否可用（True/False）

结果：

附：查询你所想安装的tensorflow对GPU和cuda的版本的对应关系：
https://www.tensorflow.org/install/source#tested_build_configurations