diff --git a/.vscode/launch.json b/.vscode/launch.json
index f21c516..0432d79 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -13,6 +13,7 @@
             "env": {
                 "NLTK_DATA": "./data/nltk",
                 "DATA_SET_SIZE": "250",
+                "XLA_FLAGS": "--xla_gpu_cuda_data_dir=/opt/cuda"
             },
             "cwd": "${workspaceFolder}",
         }