Tensorflow從開始到放棄(技術篇)

  1. 在gpu中運行
    tf.device("/gpu:1")
    有時候這個是會出問題的,即使你在有名稱爲1的gpu時。有的操做是不能支持gpu的,應該爲session添加一些配置:
    python sess=tf.InteractiveSession(config=tf.ConfigProto(allow_soft_placement=True,log_device_placement=True))
    這樣會在不支持gpu的時候自動使用cpu
  2. 查看gpu使用狀況
    nvidia-smi -l

未完待續python

相關文章
相關標籤/搜索