在我的大学里,我们有一个拥有 Tesla GPU 的集群。但是资源是由多个部门共享的,超级计算部门要求用户唯一提供一个需要在集群中运行的程序的模块/代码对象。在这种情况下,我搜索了一些有关此的信息。超级计算机有一个队列系统(这在超级计算机中通常是共享的)。据我了解,超级计算部门要求遵循这样的程序。那么,如何获取为 GPU 编译的 Keras-Theano 模型的目标代码呢?就像我需要的gcc model.c
-->制作的一样。a.out
任何其他想法都非常感谢。
最简单的解决方案应该是酸洗 theano 函数,但这只会保存优化的图形而不是生成的代码。我不确定这是否适用于您的情况。
您可以使用命令theano-cache list
查找生成的操作代码的目录,该目录通常位于/home/user/.theano/
. 但是,将其手动编译到模块中可能会很复杂。
还有一个用于共享库生成的PR,但尚未合并。