Tensorflow2.0-gpu安装及测试

作者 : icbug 发布时间: 2020-08-21 文章热度:83 共330个字,阅读需1分钟。 本文内容有更新 手机浏览

注意注意!这里需要使用指定版本安装tensorflow-2.0版本

安装命令
CPU:

pip install tensorflow == 2.0.0

CPU不需要安装前面教程的内容,直接安装即可

GPU:
需要安装前面教程的内容

pip install tensorflow-gpu==2.0.0

然后运行这段代码进行测试!

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'

a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)

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

常见问题FAQ

本站资源是否全部免费?
本站文章和普通速度下载资源免费,高速下载资源和讲解视频需要成为SVIP才能查看哦!
本站采用标准 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明!
icbug创客 » Tensorflow2.0-gpu安装及测试

发表评论

CAPTCHAis initialing...