查看安装的tensorflow是哪个版本,查看Tensorflow是GPU还是CPU

>>>import tensorflow as tf
>>>tf.__version__
'2.1.0'
>>> tf.__path__
['C:\\Users\\wei\\.virtualenvs\\pyqt51-EL5sW8oD\\lib\\site-packages\\tensorflow']
>>>

输入 import tensorflow as tf

待执行完上述语句后,输入 tf.__version__
这是查找版本号(注意,此处version的左右各有两个下划线)

输入tf.__path__,查找安装位置

查看tensorflow是否使用GPU

 tf.config.list_physical_devices('GPU')

注:只有在终端中运行代码才会显示,在编译器上却不行)


import os

from tensorflow.python.client import device_lib

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

 

if __name__ == "__main__":

    print(device_lib.list_local_devices())

  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值