Add cuda-toolkit and cuda-nvcc to Conda environment to provide nvcc for Transformer Engine compilation

cosmos-predict1.yaml

@@ -16,6 +16,8 @@ dependencies:
   # - torchaudio=2.3.1
   # - pytorch-cuda=12.1
   - cudnn
+  - cuda-toolkit=12.1
+  - cuda-nvcc=12.1
   # - libcublas
 
   - pip: