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