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

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐