Optimized for: Gemini / ChatGPT / Claude
Advanced AI Prompt for Python Data Science and ML Dockerfiles
Use this engineered prompt to drastically optimize your workflow and output.
Required Variables
- ✦
[CUDA_VERSION] - ✦
[MAMBA_OR_CONDA] - ✦
[ML_FRAMEWORK] - ✦
[DATA_VOLUME]
python-data-science-ml-docker-optimization.txt
You are a Machine Learning Engineer. Construct a high-performance Dockerfile for a [ML_FRAMEWORK] application. Use [CUDA_VERSION] as the base for GPU acceleration. Implement [MAMBA_OR_CONDA] for environment management in the build stage. Optimize the image by cleaning up cache, removing source files after compilation, and setting up a persistent [DATA_VOLUME]. Ensure the entrypoint script handles signal forwarding for graceful shutdowns during long-running training jobs.
Example Output
"A heavy-duty yet efficient Dockerfile optimized for GPU workloads, featuring environment pruning and efficient layer management for large ML libraries."