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

Question

我使用LSTM模型时遇到了不确定性问题，读到了应该在CUBLAS_WORKSPACE_CONFIG环境变量中设置单一缓冲区大小，方法如下：

    CUBLAS_WORKSPACE_CONFIG=:4096:2

我正在使用Python 3.7和cudaToolkit 11.0.221的Anaconda 1.9.12，但不知道在哪里插入该指令（除了知道它不在我的Python代码中）;-)

如有任何提示，不胜感激。 提前感谢您的帮助。

- JDE65

Answer 1

一些关于CUBLAS_WORKSPACE_CONFIG环境变量的参考可以在这里和这里找到。

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

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

例如，在您的Python代码开头添加以下内容：

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

（如果尚未导入）你可能需要import os。

- Robert Crovella

---

太好了！感谢您提供非常精确和高效的答案。已经实现了。让我们看看即使使用LSTM和Conv1D NN，确定性是否会增加;-) - JDE65

太好了，这个完美地运作了。请将其标记为已接受的答案。 - Stefan

请注意，您可能需要重新启动Python控制台才能使其正常工作。 - Stefan