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
- 上一篇 »Tensorflow如何选择GPU？
- 下一篇 »tensorflow下载和安装
import os
if Bert_Use_GPU:
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1' #使用GPU0,1
else:
os.environ['CUDA_VISIBLE_DEVICES'] = '-1' #使用CPU