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) 编辑