gpu

Dockerfile

@@ -1,9 +1,12 @@
-# Use the official Python 3.9 image
-FROM python:3.9
+# Use an official CUDA runtime as a parent image
+FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
 
 # Set the working directory to /code
 WORKDIR /code
 
+# Install Python
+RUN apt-get update && apt-get install -y python3.10 python3-pip
+
 # Copy the current directory contents into the container at /code
 COPY ./requirements.txt /code/requirements.txt
 

requirements.txt

@@ -9,5 +9,5 @@ accelerate==0.21.0
 diffusers==0.19.3
 torchvision==0.15.2
 safetensors==0.3.1
-invisible-watermark==0.2.0
-opencv-python-headless==4.8.0.74
+# invisible-watermark==0.2.0
+# opencv-python-headless==4.8.0.74