diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 50d797e..865d71d 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -26,20 +26,22 @@ jobs:
           - name: Linux (CUDA 11.8, Python 3.10, PyTorch 2.0)
             enable_cuda: true
             cuda: "11.8.0"
+            cuda_version: "118"
             gcc: "10.3.*"
             nvcc: "11.8"
             python: "3.10.*"
-            torchani: "2.2.*"
+            torchani: "2.2.4"
             pytorch: "2.0.*"
 
           # Latest supported versions (with CUDA)
           - name: Linux (CUDA 12, Python 3.13, PyTorch 2.5)
             enable_cuda: true
             cuda: "12.6.0"
+            cuda_version: "12"
             gcc: "10.3.*"
             nvcc: "12.*"
             python: "3.13.*"
-            torchani: "2.2.*"
+            torchani: "2.2.4"
             pytorch: "2.5.*"
 
     steps:
@@ -70,7 +72,8 @@ jobs:
         sed -i -e "/gxx_linux-64/c\  - gxx_linux-64 ${{ matrix.gcc }}" \
                -e "/torchani/c\  - torchani ${{ matrix.torchani }}" \
                -e "/python/c\  - python ${{ matrix.python }}" \
-               -e "/pytorch-gpu/c\  - pytorch-gpu ${{ matrix.pytorch }}" \
+               -e "/pytorch-gpu/c\  - pytorch-gpu=${{ matrix.pytorch }}=cuda${{ matrix.cuda_version }}*" \
+               -e "/cuda/c\  - cuda ${{ matrix.cuda }}" \
                environment.yml
 
     - name: Show dependency file
diff --git a/environment.yml b/environment.yml
index b22fdd1..d49fbfd 100644
--- a/environment.yml
+++ b/environment.yml
@@ -5,9 +5,9 @@ dependencies:
   - gxx_linux-64 10.3.*
   - make
   - mdtraj
-  - torchani 2.2.*
+  - torchani
   - pytest
   - python 3.10.*
   - pytorch-gpu 2.0.*
   - sysroot_linux-64 2.17
-
+  - cuda