标签: docker gpu caffe pycaffe
我正在使用caffe使用resnet 50进行图像分割。问题是网络太宽,无法容纳一个11 gb内存的GPU(批处理大小为1)。我曾想将其拆分为2 gpus,但仍然有点不工作。我在火车命令行中进行了更改,并安装了必要的软件包。我正在使用Docker容器进行培训。有人可以给我一些建议如何使其工作。非常感谢。