import os if Bert_Use_GPU: os.environ[‘CUDA_VISIBLE_DEVICES‘] = ‘0,1‘ #使用GPU0,1 else: os.environ[‘CUDA_VISIBLE_DEVICES‘] = ‘-1‘ #使用CPU
tensorflow如何切换CPU和GPU
原文：https://www.cnblogs.com/LiuXinyu12378/p/13626622.html
