会员
周边
捐助
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
代码空间
Computer Vision/Machine Learning/Evolutionary Computation...
博客园
首页
新随笔
联系
订阅
管理
2018年12月31日
Keras/Tensorflow选择GPU/CPU运行
摘要: 首先,导入os,再按照PCI_BUS_ID顺序,从0开始排列GPU, 然后就可以选择用哪一个或者那几个GPU运行: 用0号GPU,即'/gpu:0'运行; 用0号和1号设备,即'/gpu:0'、'/gpu:1'运行; 用7号设备,即'/gpu:7'运行,但是'/gpu:7'根本不存在,于是就只能用C
阅读全文
posted @ 2018-12-31 15:28 MSTK
阅读(4051)
评论(0)
推荐(0)
编辑
公告