Skip to content

Commit

Permalink
Merge pull request #35 from crufter/fix-multigpu
Browse files Browse the repository at this point in the history
Fix docker build for multi-gpu
  • Loading branch information
NickLucche authored Jun 21, 2024
2 parents 3b0531d + 34e6464 commit b4e0bc0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
FROM pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime
WORKDIR /app
COPY *.py requirements.txt /app
COPY *.py requirements.txt *.pickle /app
RUN apt update && apt install -y git
RUN pip install -r requirements.txt
ENV GRADIO_SERVER_PORT=7860
Expand Down

0 comments on commit b4e0bc0

Please sign in to comment.