caffe2多次载入模型问题，在django后端中每次有用户访问时载入一个模型，第一个可以，之后报错

Question

我使用django搭建了一个后端服务器，处理caffe2对rtsp视频流进行处理。当每有一个
用户访问时都会载入一次模型，对视频流中的目标就行识别，当第一个用户访问时可以
正常识别使用，但是当第二个用户访问时，多个用户同时使用就会报错，不知道是什么原因
RuntimeError: [enforce fail at batch_permutation_op.cu:63] X.dim32(0) == indices.dim32(0). X.dim32(0) must be equal to indices.dim32(0)(1001 vs. 1000)Error from operator:
input: "gpu_0/roi_feat_shuffled" input: "gpu_0/rois_idx_restore_int32" output: "gpu_0/roi_feat" name: "" type: "BatchPermutation" device_option { device_type: 1 cuda_gpu_id: 0
}

有没有可能与workspaceyou'guan

Answer 1

你解决了吗，是不是一块GPU不能并行操作，做个任务栈一个一个处理就可以了，或者不同的显卡