小编JDE*_*E65的帖子

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

我的 LSTM 模型存在不确定性问题,我读到应该在 CUBLAS_WORKSPACE_CONFIG 环境变量中设置单个缓冲区大小,其中:

    CUBLAS_WORKSPACE_CONFIG=:4096:2
Run Code Online (Sandbox Code Playgroud)

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

任何指示都是非常受欢迎的。在此先感谢您的帮助

python cuda

6
推荐指数
1
解决办法
3800
查看次数

标签 统计

cuda ×1

python ×1