我有一个pkl文件,其中包含我最初使用GPU转储的ndarray。我用GPU解压它,现在我想把它存储为任何东西,以后我可以用CPU解压。我在一台超级计算机上运行所有的东西,后来我只想在一台普通的计算机上访问ndarray,而不需要一个漂亮的GPU。我研究了诸如
np.save()
np.savez()
但是save()
我不能设置allow_pickle=False
,当我加载用savez()
存储的数组时,它是空的。在
这就是我保存东西的方法:
我运行THEANO_FLAGS="device=gpu,floatX=float32" srun -u python deep_q_rl/unpicklestuff.py
在unpicklestuff.py公司名称:
^{pr2}$这是我以后加载它们的方式:
q_layers = np.load(path)
目前没有回答
相关问题 更多 >
编程相关推荐