Kangarroar commited on
Commit
ecc6932
1 Parent(s): 3ebffc6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -4,6 +4,7 @@ WORKDIR /app
4
 
5
  COPY ./requirements.txt /app/requirements.txt
6
  COPY ./packages.txt /app/packages.txt
 
7
  RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
8
  RUN pip3 install --no-cache-dir -r /app/requirements.txt
9
 
 
4
 
5
  COPY ./requirements.txt /app/requirements.txt
6
  COPY ./packages.txt /app/packages.txt
7
+ 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
8
  RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
9
  RUN pip3 install --no-cache-dir -r /app/requirements.txt
10