Kangarroar commited on
Commit
7d201e4
1 Parent(s): 69d0bb2

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -11,6 +11,8 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
11
  git \
12
  ffmpeg \
13
  && apt-get clean && rm -rf /var/lib/apt/lists/*
 
 
14
  COPY ./requirements.txt /app/requirements.txt
15
  COPY ./packages.txt /app/packages.txt
16
  RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
 
11
  git \
12
  ffmpeg \
13
  && apt-get clean && rm -rf /var/lib/apt/lists/*
14
+
15
+ WORKDIR $HOME/app
16
  COPY ./requirements.txt /app/requirements.txt
17
  COPY ./packages.txt /app/packages.txt
18
  RUN pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html