TensorFlow 安装与环境配置

技术分享 手册留言
4

  1. 可以分别使用 GPU 和 CPU 运行同一个模型来选择合适的运算设备。GPU 和 CPU 的选择与模型类型也有关系，例如如果是卷积神经网络则 gtx1050 可能胜过 i7，但如果是强化学习则不一定。

  2. 可以简单地认为，是的。不需要两个版本都装，TensorFlow 2.1 默认安装（pip install tensorflow）就已经同时支持 CPU 和 GPU，可以使用

    cpus = tf.config.list_physical_devices (device_type=‘CPU’)
    tf.config.set_visible_devices (devices=cpus)

来限定只使用 CPU 进行运算。