确定性问题：在 CUBLAS_WORKSPACE_CONFIG 环境变量中设置缓冲区大小

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对环境变量的一些引用位于此处和此处。

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

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

在你的Python代码的早期，类似这样的东西：

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

import os（如果还没有的话，您可能需要）