Add Docker Nvidia CUDA Hardware Acceleration

Dockerfile_nvidia_cuda_cudnn_gpu

@@ -0,0 +1,14 @@
+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
+
+WORKDIR /rembg
+
+RUN apt-get update && apt-get install -y --no-install-recommends python3-pip python-is-python3 curl && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+COPY . .
+
+RUN python -m pip install ".[gpu,cli]" --break-system-packages
+RUN rembg d
+
+EXPOSE 7000
+ENTRYPOINT ["rembg"]
+CMD ["--help"]