如何使用 VS Code 扩展增加 Jupyter Notebook 内存限制？

Question

问题

我正在处理大型数据结构，但我发现遇到了以下错误。例如，运行此代码：

np.zeros((6867151, 20, 4))

导致错误提示：

---------------------------------------------------------------------------
MemoryError                               Traceback (most recent call last)
Cell In[64], line 1
----> 1 np.zeros((6867151, 20, 4))

MemoryError: Unable to allocate 4.09 GiB for an array with shape (6867151, 20, 4) and data type float64

我知道我遇到了人为限制，因为我的电脑有 32 GB RAM，但运行时仅使用了 32%。

我尝试过的

这个问题似乎之前已得到解答，但他们似乎没有使用 Jupyter Notebook 的 VS Code 扩展（就像我一样）。因此，我不太确定如何解决这个问题。谢谢。

Answer 1

您可以尝试添加以下代码

import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"