tensorflow 常用命令

常用命令

1. 查看tensorboard

   在estimator模式下查看tensorboard
   tensorboard --logdir model_dir --host 0.0.0.0 --port 8888

2. 指定gpu

   import os
   os.environ["CUDA_VISIBLE_DEVICES"] = "3"

3. 查看gpu使用情况

   nvidia-smi

4. 查看tensorflow版本

   import tensorflow as tf
   print(tf.__version__)

5. 查看cuda版本

   cat /usr/local/cuda/version.txt
   CUDA Version 8.0.44

总结

1. https://github.com/c00h00g/tf_chg/