马春杰杰博客
致力于深度学习经验分享!

[mcj]测试TensorFlow-GPU是否安装成功

然后测试TensorFlow-GPU是否可用:

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

如果可用的话,会输出GPU信息:

 

如果你对这篇文章有什么疑问或建议,欢迎下面留言提出,我看到会立刻回复!

打赏
未经允许不得转载:马春杰杰 » [mcj]测试TensorFlow-GPU是否安装成功

留个评论吧~ 抢沙发 评论前登陆可免验证码!

私密评论

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址(选填,便于回访^_^)
切换注册

登录

忘记密码 ?

切换登录

注册