diff --git a/docker/GPUDockerfile b/docker/GPUDockerfile index 8b5b96d..982c2c0 100644 --- a/docker/GPUDockerfile +++ b/docker/GPUDockerfile @@ -5,8 +5,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \ curl python3-pip -RUN pip3 install --upgrade pip && \ - pip3 install torch==1.13.1 xformers==0.0.16rc425 +RUN pip3 install --upgrade pip +RUN pip3 install torch==1.13.1 +RUN pip3 install https://files.pythonhosted.org/packages/6a/c8/f5c62870f9c3c4c9a87dc260869894860e2f33552addec83bdb4cda1e315/xformers-0.0.16-cp38-cp38-manylinux2014_x86_64.whl --platform manylinux2014_x86_64 --no-deps --dry-run ARG version