最近在跑FCN的时候发现caffe.set_device()只能有一个int参数,导致只能用一个GPU跑,于是开始着手解决。

caffe声明:https://github.com/BVLC/caffe/blob/master/docs/multigpu.md

截图如下:

Multi-GPU Usage:caffe下的多GPU同时使用

多GPU接口只支持C/C++,以及训练。

正在绞尽脑汁解决~

相关文章:

  • 2021-08-15
  • 2021-05-29
  • 2021-07-14
  • 2022-12-23
  • 2021-08-02
  • 2021-11-24
  • 2021-12-08
  • 2022-01-29
猜你喜欢
  • 2022-12-23
  • 2022-01-02
  • 2021-11-14
  • 2021-10-13
  • 2021-07-30
  • 2022-12-23
  • 2022-12-23
相关资源
相似解决方案