Fix PyTorch verification command in Dockerfile using heredoc for robust multi-line Python

Dockerfile

@@ -57,16 +57,18 @@ RUN echo "Verifying Python and Conda installations..."
 RUN python --version
 RUN conda env list
 RUN echo "Verifying PyTorch and CUDA availability..."
-# REVISED: Break into more explicit separate statements with newlines for robustness
-RUN conda run -n cosmos-predict1 python -c "\
-import torch;\n\
-print('PyTorch Version: ' + torch.__version__);\n\
-print('CUDA Available: ' + str(torch.cuda.is_available()));\n\
-if torch.cuda.is_available():\n\
-    print('CUDA Device Name: ' + torch.cuda.get_device_name(0));\n\
-else:\n\
-    print('CUDA Device Name: N/A');\n\
-" || echo "PyTorch verification failed. Check dependencies in cosmos-predict1.yaml."
+# REVISED AGAIN: Use a heredoc for multi-line Python code
+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
+# Add the fallback echo if the python command fails
+RUN [ $? -eq 0 ] || echo "PyTorch verification failed. Check dependencies in cosmos-predict1.yaml."
 # --- End Verification Steps ---
 
 # Make the start.sh script executable.