Adding CUBLAS_WORKSPACE_CONFIG:4096:8 to Vscode launch.json file

vision
1

This is mostly related to VsCode but since I am running a deep learning model in PyTorch and the issue is related to making PyTorch functions deterministic, I ask my question here. Hope you guys’ experiences can help me.

I need to add one of these two environemtal variables before running my code in vscode:

CUBLAS_WORKSPACE_CONFIG=:4096:8 or CUBLAS_WORKSPACE_CONFIG=:16:8

yet I don’t know exactly add this to the launch.json file of vscode program.
I tried this:

{
“version”: “0.2.0”,
“configurations”: [

    {
        "name": "Python: Current File",
        "type": "python",
        "request": "launch",
        "program": "${file}",
        "console": "integratedTerminal",
        "args": ["${CUBLAS_WORKSPACE_CONFIG:4096:8}",

to no avail. I appreciate your guidance here.