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

related bug: https://github.com/AbdBarho/stable-diffusion-webui-docker/issues/694
This commit is contained in:
Maifee Ul Asad 2024-05-28 16:47:24 +06:00 committed by GitHub
parent 35a18b3d46
commit 3dc0217730
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -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