- 博客(2)
- 资源 (4)
- 收藏
- 关注
RSS订阅原创 深度学习【13】tensorflow保存graph和参数为pb文件
from tensorflow.python.framework.graph_util import convert_variables_to_constants graph = convert_variables_to_constants(sess, sess.graph_def, ["out"]) #out为保存网络的最后输出节点名称 tf.train.write_graph(grap...
2017-05-24 14:28:36
7910
原创 深度学习【12】DL平台GPU分配
本说明只适用于GPU显存分配，以及如何使用指定使用某个GPU。 要同时使用多GPU计算时，tensorflow需要自己修改代码，theano需要安装libgpuarrary（还没安装成功） theano： 例子： THEANO_FLAGS='device=gpu0,lib.cnmem=0.1' python test.py 说明： device=gpu0,表示使用ID号为0的G...
2017-05-23 12:00:20
1867
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人