检查安装的tensorflow版本信息

    科技2024-03-24  78

    1、tensorflow信息

    tf.__version__  # 版本信息 tf.__path__  # 安装位置

    2、是否是GPU

    tf.test.is_gpu_available()   # 输出为True则为GPU

    3、选择GPU还是CPU

           整体来说,代码在这两种环境下运行几乎没什么区别,但GPU更加适合复杂密集的高速运算,像入手深度学习进行模型训练首选GPU。

    Processed: 0.023, SQL: 8