Removed header symlinks before transformer-engine install

Dockerfile

@@ -49,37 +49,10 @@ RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
         torchaudio==2.3.1 \
         --index-url https://download.pytorch.org/whl/cu121
 
-# --- NEW SECTION: Patch Transformer Engine linking issues by creating symlinks for headers ---
-# These commands must run AFTER conda env creation and BEFORE transformer-engine pip install.
-RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
-    conda activate cosmos-predict1 && \
-    echo "Creating symlinks for NVIDIA headers..." && \
-    ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/ && \
-    ln -sf $CONDA_PREFIX/lib/python3.10/site-packages/nvidia/*/include/* $CONDA_PREFIX/include/python3.10/
-# --- END NEW SECTION ---
-
 # Install Transformer Engine separately after PyTorch and cuDNN are in place and headers are linked.
 RUN . $CONDA_DIR/etc/profile.d/conda.sh && \
     conda activate cosmos-predict1 && \
     pip install transformer-engine[pytorch]==1.12.0
-    #pip install --no-cache-dir --no-build-isolation transformer-engine[pytorch]==1.12.0
-
-# --- Verification Steps ---
-RUN echo "Verifying Python and Conda installations..."
-RUN python --version
-RUN conda env list
-RUN echo "Verifying PyTorch and CUDA availability..."
-RUN conda run -n cosmos-predict1 python <<EOF
-import torch
-print('PyTorch Version: ' + torch.__version__)
-print('CUDA Available: ' + str(torch.cuda.is_available()))
-if torch.cuda.is_available():
-    print('CUDA Device Name: ' + torch.cuda.get_device_name(0))
-else:
-    print('CUDA Device Name: N/A')
-EOF
-RUN [ $? -eq 0 ] || echo "PyTorch verification failed. Check dependencies in cosmos-predict1.yaml."
-# --- End Verification Steps ---
 
 # Make the start.sh script executable.
 RUN chmod +x /app/start.sh