From 99a7dc4a9a39eccfad09693d2570d719d1567aac Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Wed, 1 Nov 2023 09:51:26 +0800
Subject: [PATCH] update GPUDockerfile

---
 docker/GPUDockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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