SoSa123456 commited on
Commit
ac7c4d1
1 Parent(s): 50c6ac6

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ ADD . /app
10
  # Install any needed packages specified in requirements.txt
11
  RUN python -m pip install --upgrade pip
12
  RUN pip install -r requirements.txt
13
-
14
  # Make port 80 available to the world outside this container
15
  EXPOSE 80
16
 
 
10
  # Install any needed packages specified in requirements.txt
11
  RUN python -m pip install --upgrade pip
12
  RUN pip install -r requirements.txt
13
+ RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
14
  # Make port 80 available to the world outside this container
15
  EXPOSE 80
16