指定GPU

  • 终端指定
1
CUDA_VISIBLE_DEVICES=0 nohup python demo.py >> base_log.out 2>&1 &
  • 程序指定
1
2
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

TensorFlow常用操作

检测GPU是否可用

1
2
import tensorflow as tf
tf.test.is_gpu_available()