cyberspyde commited on
Commit
63bd0c7
1 Parent(s): 1203715
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -6,6 +6,8 @@ COPY ./requirements.txt /code/requirements.txt
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
 
 
9
  RUN apt-get -y update
10
 
11
  RUN useradd -m -u 1000 user
 
6
 
7
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
8
 
9
+ RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
10
+
11
  RUN apt-get -y update
12
 
13
  RUN useradd -m -u 1000 user