Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

🐳 Docker Configuration: Set up PyTorch GPU Environment #11

Open
wants to merge 5 commits into
base: main
Choose a base branch
from

Conversation

Sunwood-ai-labs
Copy link

📋 Overview

This PR introduces Docker configuration for the PyTorch application with GPU support, enabling containerized deployment and easier environment management.

🔄 Key Changes

  • Create Dockerfile with PyTorch 2.2.1 and CUDA 12.1 support
  • Configure docker-compose.yml with GPU capabilities
  • Set up appropriate ignore files for Git and Docker
  • Enable share mode in Gradio application

🔧 Technical Details

  • Base image: pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
  • Configured NVIDIA GPU integration
  • Set up volume mounts for application code and model caching
  • Added proper exclusions for development artifacts and cache files
  • Modified Gradio launch configuration for remote access

- Exclude unnecessary files and directories from Docker context
- Prevent copying of development artifacts and cache files
- Optimize build process and reduce image size
- Add Gradio certificate and cache directory exclusions
- Ensure sensitive and temporary files are not tracked
- Set up PyTorch 2.2.1 with CUDA 12.1 support
- Configure system dependencies and Python packages
- Establish working directory and application deployment structure
- Modify launch configuration to allow remote access
- Enable sharing capability for wider accessibility
- Configure service with NVIDIA GPU integration
- Set up volume mounts for application code and model caching
- Define port mapping and resource allocations
- Enable GPU capabilities for ML operations
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant