# 用cpu
caffe.set_mode_cpu()
# 用gpu
caffe.set_device(0)
caffe.set_mode_gpu()
caffe matlab
最新推荐文章于 2021-03-16 11:07:07 发布
博客介绍了Caffe框架中设置CPU和GPU模式的代码。使用caffe.set_mode_cpu()可设置为CPU模式,使用caffe.set_device(0)和caffe.set_mode_gpu()可设置为GPU模式。
1108

被折叠的 条评论
为什么被折叠?



