Tensorflow 指定GPU

import os

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

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

你可能感兴趣的:(Machine,Learning)