Revert to pytorch 1.0.1

kausmik · May 4, 2019 · 9e014a8 · 9e014a8
1 parent e9c84c7
commit 9e014a8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -7,12 +7,12 @@ COPY requirements.txt .
 
 RUN pip install --upgrade pip
 
-RUN pip install --no-cache-dir -r requirements.txt --upgrade
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY app app/
 
 RUN python app/server.py
 
 EXPOSE 5042
 
-CMD ["python", "app/server.py", "serve"]
+CMD ["python", "app/server.py", "serve"]
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 numpy
 torchvision
-https://download.pytorch.org/whl/cpu/torch-1.1.0-cp37-cp37m-linux_x86_64.whl
-fastai==1.0.52
+https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp37-cp37m-linux_x86_64.whl
+fastai
 starlette
 uvicorn==0.3.32
 python-multipart