如何安装并查看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