import os
if Bert_Use_GPU:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' #使用GPU0,1
else:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #使用CPU
相关文章：
猜你喜欢
相关资源
相似解决方案
热门标签
import os
if Bert_Use_GPU:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' #使用GPU0,1
else:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #使用CPU
相关文章：