[docker]: update to cuda-toolkit 12.1 in service/invoke/Dockerfile

related bug: AbdBarho#694
maifeeulasad · May 28, 2024 · 3dc0217 · 3dc0217
1 parent 35a18b3
commit 3dc0217
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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