美文网首页
Tensorflow-GPU测试是否可用

Tensorflow-GPU测试是否可用

作者: 廿怎么念 | 来源:发表于2020-06-13 15:11 被阅读0次

新版tensorflow-gpu, 2.2.0 版推荐的测试程序与之前的1.2.0 不同,记录一下。

  1. Tensorflow-gpu 1.x.x, 如Tensorflow-gpu 1.2.0, 可使用以下代码
import tensorflow as tf
tf.test.is_gpu_available()

(1) jupyter notebook示例


返回为True, 表示成功

(2)pycharm示例


加载成功,找到可用显卡
  1. Tensoeflow-gpu 2.x.x,如Tensorflow-gpu 2.2.0, 可使用以下代码
import tensorflow as tf
tf.config.list_physical_devices('GPU')

(1) jupyter notebook


加载成功,找到可用显卡

(2) pycharm 测试示例


成功加载,找到可用显卡

相关文章

网友评论

      本文标题:Tensorflow-GPU测试是否可用

      本文链接:https://www.haomeiwen.com/subject/njnatktx.html