Bump pytorch containers (#697)

Closes #696
Closes #694
This commit is contained in:
AbdBarho
2024-05-28 19:39:33 +02:00
committed by GitHub
parent 35a18b3d46
commit f1bf3b0943
3 changed files with 5 additions and 5 deletions

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-runtime
ENV DEBIAN_FRONTEND=noninteractive PIP_EXISTS_ACTION=w PIP_PREFER_BINARY=1