Skip to content

Commit b4e0bc0

Browse files
authored
Merge pull request #35 from crufter/fix-multigpu
Fix docker build for multi-gpu
2 parents 3b0531d + 34e6464 commit b4e0bc0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
22
WORKDIR /app
3-
COPY *.py requirements.txt /app
3+
COPY *.py requirements.txt *.pickle /app
44
RUN apt update && apt install -y git
55
RUN pip install -r requirements.txt
66
ENV GRADIO_SERVER_PORT=7860

0 commit comments

Comments
 (0)