Tensorflow 指定GPU

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/vernice/article/details/82413660

import os

os.environ["CUDA_VISIBLE_DEVICES"]="0"

仅仅GPU 0 可见，process在GPU 0 上运行