python - 确定性问题 : set a buffer size in the CUBLAS_WORKSPACE_CONFIG environmental variable

Question

我有一个 LSTM 模型的非确定性问题，我读到我应该在 CUBLAS_WORKSPACE_CONFIG 环境变量中设置一个缓冲区大小，其中:

    CUBLAS_WORKSPACE_CONFIG=:4096:2

我正在使用 Anaconda 1.9.12 和 Python 3.7，以及 cudaToolkit 11.0.221。我不知道在哪里插入该指令(除了我知道它不在我的 python 代码中；-)

欢迎任何指示。预先感谢您的帮助

Answer 1

一些对 CUBLAS_WORKSPACE_CONFIG 环境变量的引用是 here和 here .

设置环境变量通常取决于您使用的操作系统，例如 windows或 linux .

但是，从您的 python 脚本中，应该可以使用类似于设置 CUDA_VISIBLE_DEVICES 环境变量的方法，描述 here .

像这样的东西，很早就在你的 python 代码中:

os.environ["CUBLAS_WORKSPACE_CONFIG"]=":4096:2"

(如果还没有，您可能需要import os)