Add torchvision nightly

kausmik · Jan 24, 2019 · 01ef9fa · 01ef9fa
1 parent 9fbb5d3
commit 01ef9fa
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -7,7 +7,7 @@ COPY requirements.txt .
 
 RUN pip install --upgrade pip
 
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt --upgrade
 
 COPY app app/
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,7 @@
--f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
-torch_nightly
 numpy
+torchvision_nightly
+torch_nightly
+-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
 git+git://github.com/fastai/fastai.git@9835e1805d0f9609c28a3258200e2824240e3deb#egg=fastai
 starlette
 uvicorn