From 3dc0217730d930d8d6cc165306d9f017389f85b6 Mon Sep 17 00:00:00 2001
From: Maifee Ul Asad <29339330+maifeeulasad@users.noreply.github.com>
Date: Tue, 28 May 2024 16:47:24 +0600
Subject: [PATCH] [docker]: update to cuda-toolkit 12.1 in
 `service/invoke/Dockerfile`

related bug: https://github.com/AbdBarho/stable-diffusion-webui-docker/issues/694
---
 services/invoke/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/services/invoke/Dockerfile b/services/invoke/Dockerfile
index 209c368be..b0c80e06f 100644
--- a/services/invoke/Dockerfile
+++ b/services/invoke/Dockerfile
@@ -3,7 +3,7 @@ RUN apk add --no-cache aria2
 RUN aria2c -x 5 --dir / --out wheel.whl 'https://github.com/AbdBarho/stable-diffusion-webui-docker/releases/download/6.0.0/xformers-0.0.21.dev544-cp310-cp310-manylinux2014_x86_64-pytorch201.whl'
 
 
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-devel
 
 ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1