擅长:python、mysql、java
<p>我不认为你能用python接口在<a href="/questions/tagged/caffe" class="post-tag" title="show questions tagged 'caffe'" rel="tag">caffe</a>中做到这一点。<br/>
但是我认为可以用c++来实现:在c++中,您可以访问<code>Blob</code>的{<cd2>}。您可以编写在设备上运行的代码,并直接从gpu“填充”输入Blob的<code>mutable_gpu_data()</code>。一旦你做了这个更新,caffe应该可以继续它的<code>net->forward()</code>。在</p>
<p><strong>更新</strong><br/>
2017年9月19日,<a href="https://github.com/BVLC/caffe/pull/5904" rel="nofollow noreferrer">PR #5904</a>合并为master。此PR通过python接口公开blob的GPU指针。<br/>
您可以直接从python<strong>访问<code>blob._gpu_data_ptr</code>和<code>blob._gpu_diff_ptr</code>,风险自负</strong>。在</p>