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

我对LSTM模型存在不确定性问题，我读到我应该在CUBLAS_WORKSPACE_CONFIG环境变量中设置一个缓冲区大小，使用：

    CUBLAS_WORKSPACE_CONFIG=:4096:2

我将Anaconda1.9.12与Python3.7和cudaToolkit 11.0.221一起使用。我不知道在哪里插入该指令（除了我知道它不在我的python代码中；-）

任何迹象都是非常受欢迎的。

---

提前感谢您的帮助

一些对

CUBLAS\u WORKSPACE\u CONFIG

环境变量的引用是和

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

但是，在python脚本中，应该可以使用类似于设置所描述的

CUDA\u VISIBLE\u DEVICES

环境变量的方法

类似这样的内容，在python代码的早期：

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

---

（如果还没有导入操作系统，您可能需要导入操作系统）

太好了！Thnks需要非常精确和有效的答案。它已经实现了。让我们看看即使使用LSTM和Conv1D NN，确定性是否也会增加；-）太好了，这个很好用。请将其标记为已接受答案。注意，您可能需要重新启动python控制台才能使其工作。