diff --git a/.github/workflows/notebook-run.yml b/.github/workflows/notebook-run.yml
index 5c66171e..0b71c637 100644
--- a/.github/workflows/notebook-run.yml
+++ b/.github/workflows/notebook-run.yml
@@ -9,6 +9,7 @@ on:
         default: '11-8'
 
 env:
+  cudaVersion: ${{github.event.inputs.cudaVersion}}
 
 jobs:
   build_env: