diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index ffbc25ea..600dd095 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -37,6 +37,7 @@ steps:
 # pushd src && make && make install && popd
 - script: |
     source activate pp_env
+    export CUDA_HOME=/usr/local/cuda-12.6
     export PATH=${CUDA_HOME}/bin${PATH:+:${PATH}}
     export LD_LIBRARY_PATH=${CUDA_HOME}/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
     export SKETCHLIB_INSTALL=azure
@@ -45,6 +46,7 @@ steps:
 
 - script: |
     source activate pp_env
+    export CUDA_HOME=/usr/local/cuda-12.6
     export PATH=${CUDA_HOME}/bin${PATH:+:${PATH}}
     export LD_LIBRARY_PATH=${CUDA_HOME}/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
     cd test && python run_test.py