TensorFlow2.x,GPU代码测试

爱听许嵩歌

已于 2022-02-19 13:26:05 修改

阅读量2.9w

点赞数 18

分类专栏： Tensorflow2.x学习文章标签： TensorFlow2.0 GPU代码测试

于 2019-11-06 11:05:50 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_45092662/article/details/102931156

版权

Tensorflow2.x学习专栏收录该内容

23 篇文章

订阅专栏

本文介绍如何使用简单的代码测试TensorFlow2.x的GPU功能是否正常工作，并提供安装教程链接，包括如何通过conda安装TensorFlow和PyTorch的GPU版本。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TensorFlow2.x,GPU代码测试

代码如下

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'  # 不显示等级2以下的提示信息

print('GPU', tf.test.is_gpu_available())

a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)

结果如下就证明tensorflow-gpu安装成功了

GPU True
tf.Tensor(6.0, shape=(), dtype=float32)

2020年10月11号更新
简单安装tensorflow-gpu=2.2.0教程：https://blog.youkuaiyun.com/weixin_45092662/article/details/106980282

利用conda简单安装tensorflow和pytorch的GPU版本教程：
https://blog.youkuaiyun.com/weixin_45092662/article/details/105928951

有用请点个赞！！
本站所有文章均为原创，欢迎转载，请注明文章出处：https://blog.youkuaiyun.com/weixin_45092662。百度和各类采集站皆不可信，搜索请谨慎鉴别。技术类文章一般都有时效性，本人习惯不定期对自己的博文进行修正和更新，因此请访问出处以查看本文的最新版本。