tensorflow测试程序

对着背影说爱祢 提交于 2020-03-04 19:20:45

终于安装成功了。。。纪念一下
版本:
cuda:10.2
cuDNN:7.6.5
tensorflow:2.0.0

这些都安装成功后打开%CUDA_PATH%\bin
然后将cudart64_102.dll改名为cudart64_100.dll即可

测试程序:

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())

结果:
在这里插入图片描述

开心!!!~~

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!