如何将一个Python ndarray存储到磁盘上?

2024-10-02 00:44:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个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)

Tags: 文件pyfalsegpusave计算机npcpu

热门问题