diff --git a/.vim/nvim.lua b/.vim/nvim.lua
index e07e4e5..272c22d 100644
--- a/.vim/nvim.lua
+++ b/.vim/nvim.lua
@@ -48,6 +48,11 @@ nvim_lsp.rust_analyzer.setup({
           'cargo', 'clippy', '--workspace', '--message-format=json',
           '--all-targets', '--all-features'
         }
+      },
+      runnables = {
+        extraEnv = {
+          TORCH_CUDA_VERSION = "cu121",
+        }
       }
     }
   }