Tensorflow

来自Shiyin's note
跳到导航 跳到搜索
  • 安装的时候要注意版本,查好对应的cuda版本 pip install tensorflow tensorflow-gpu [1]
  • tf.config.list_physical_devices('GPU')