点此进入CSDN

点此添加QQ好友 加载失败时会显示




2020年9月7日
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 阅读全文
posted @ 2020-09-07 14:16 高颜值的殺生丸 阅读(578) 评论(0) 推荐(0) 编辑

作者信息

昵称：

刘新宇

园龄：4年6个月


粉丝：1209


QQ：522414928