kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

update GPUDockerfile

Browse Source
This commit is contained in:
Qing 2023-11-01 09:51:26 +08:00
parent 3a90ab3da7
commit 99a7dc4a9a
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
docker/GPUDockerfile
View File

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