diff --git a/install.sh b/install.sh index 0e1cd07..75bc604 100644 --- a/install.sh +++ b/install.sh @@ -1,2 +1,2 @@ -virtualenv .env && source .env/bin/activate && pip3 install -r requirements.txt --user +virtualenv .env && source .env/bin/activate && pip3 install -r requirements.txt diff --git a/requirements.txt b/requirements.txt index a2d33cd..e54b44d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,2 @@ -https://download.pytorch.org/whl/cpu/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl -torchvision +torch==1.0.1.post2 +torchvision==0.2.2.post3