Create Dockerfile

Dockerfile

@@ -0,0 +1,21 @@
+# Use an official NVIDIA base image with PyTorch, CUDA, and cuDNN pre-installed
+# We choose a version with Python 3.11 to match your previous environment
+FROM nvcr.io/nvidia/pytorch:24.04-py3
+
+# Set the working directory
+WORKDIR /repository
+
+# Copy the requirements file into the container
+COPY requirements.txt .
+
+# Install your Python dependencies
+# Using --no-cache-dir is good practice for smaller image sizes
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of your application code (e.g., your handler.py)
+COPY . .
+
+# (Optional) Set an environment variable if needed by your platform
+# ENV TRANSFORMERS_CACHE="/repository/cache"
+
+# Your inference platform will typically handle the CMD or ENTRYPOINT