Resolve Conda environment conflicts by simplifying CUDA dependencies and using stable PyTorch

cosmos-predict1.yaml

@@ -11,14 +11,17 @@ dependencies:
   - ninja
   - gcc=12.4.0
   - gxx=12.4.0
-  - cuda=12.4
-  - cuda-nvcc=12.4
-  - cuda-toolkit=12.4
+# REMOVE these explicit CUDA dependencies from here.
+  # Let pytorch-cuda manage its own CUDA libraries.
+  # - cuda=12.4
+  # - cuda-nvcc=12.4
+  # - cuda-toolkit=12.4
   # Add PyTorch and Torchvision
-  - pytorch=2.5.0a0 
-  - torchvision
-  - torchaudio
+  - pytorch=2.3.1
+  - torchvision=0.18.1
+  - torchaudio=2.3.1
   - pytorch-cuda=12.1 
+
   # Add pip dependencies directly in the YAML
   - pip:
       - -r file:gui/requirements.txt