diff --git a/.github/workflows/conda-environments/linux-environment.yml b/.github/workflows/conda-environments/linux-environment.yml
index e70debac..19d9c236 100644
--- a/.github/workflows/conda-environments/linux-environment.yml
+++ b/.github/workflows/conda-environments/linux-environment.yml
@@ -8,7 +8,7 @@ dependencies:
   - pytorch
   - torchvision
   - torchaudio
-  - cudatoolkit=12.1
+  - pytorch-cuda=12.1
   - pip:
     # comfyui requirements
     - einops