tensorflow如何切换CPU和GPU
import os
if Bert_Use_GPU:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' #使用GPU0,1
else:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #使用CPU
多思考也是一种努力，做出正确的分析和选择，因为我们的时间和精力都有限，所以把时间花在更有价值的地方。
点此进入CSDN
import os
if Bert_Use_GPU:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' #使用GPU0,1
else:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #使用CPU
