diff --git a/.github/workflows/cuda_ci.yaml b/.github/workflows/cuda_ci.yaml
index 2ca3a3cb..4f18967a 100644
--- a/.github/workflows/cuda_ci.yaml
+++ b/.github/workflows/cuda_ci.yaml
@@ -24,14 +24,6 @@ jobs:
     - name: Get current date
       id: date
       run: echo "date=$(date +%Y-%m-%d)" >> "${GITHUB_OUTPUT}"
-    - name: Install Conda environment from environment.yml
-      uses: mamba-org/setup-micromamba@v1
-      with:
-        micromamba-version: '1.4.6-0'
-        environment-file: environment.yml
-        # persist on the same day.
-        cache-environment-key: environment-${{ steps.date.outputs.date }}
-        cache-downloads-key: downloads-${{ steps.date.outputs.date }}
     - name: Install nvcc
       uses: Jimver/cuda-toolkit@v0.2.19
       id: cuda-toolkit
@@ -41,6 +33,14 @@ jobs:
         echo "Installed cuda version is: ${{steps.cuda-toolkit.outputs.cuda}}"
         echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
         nvcc -V
+    - name: Install Conda environment from environment.yml
+      uses: mamba-org/setup-micromamba@v1
+      with:
+        micromamba-version: '1.4.6-0'
+        environment-file: environment.yml
+        # persist on the same day.
+        cache-environment-key: environment-${{ steps.date.outputs.date }}
+        cache-downloads-key: downloads-${{ steps.date.outputs.date }}
     - name: Install and run_test.py
       shell: bash -l {0}
       run: |
diff --git a/azure-pipelines.yml b/azure-pipelines.yml
index bf11cb28..1f698363 100644
--- a/azure-pipelines.yml
+++ b/azure-pipelines.yml
@@ -27,7 +27,7 @@ steps:
     wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
     sudo dpkg -i cuda-keyring_1.1-1_all.deb
     sudo apt-get update
-    sudo apt-get -y install cuda=11.2.2-1
+    sudo apt-get -y install cuda-toolkit=11.2.2-1
   displayName: Install nvcc
 
 - bash: conda env create --file environment.yml