Attempt to fix torchvision::nms error by adding cudnn and libcublas to conda dependencies

cosmos-predict1.yaml

@@ -16,15 +16,17 @@ dependencies:
   - torchvision=0.18.1
   - torchaudio=2.3.1
   - pytorch-cuda=12.1
+  - cudnn 
+  - libcublas 
 
   - pip:
       - -r file:///app/gui/requirements.txt
       - uvicorn[standard]
       - fastapi
-      - git+https://github.com/microsoft/MoGe.git 
+      - git+https://github.com/microsoft/MoGe.git
       - einops
       - omegaconf
       - diffusers
       - megatron
       - IPython
-      - megatron.core
+      - megatron.core