Skip to content
Snippets Groups Projects
Unverified Commit c67cf3c4 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Fix CI by limiting the version of torchvision (#2005)

* try pytorch 1.4 instead of 1.2

* use pytorch 1.2 and torchvision 0.4.0
parent f4da38a7
No related branches found
No related tags found
No related merge requests found
......@@ -24,8 +24,8 @@ before_install:
- PATH=${CUDA_HOME}/bin:${PATH}
install:
- pip install Cython torch==1.2
- pip install Pillow==6.2.2
- pip install Pillow==6.2.2 # remove this line when torchvision>=0.5
- pip install Cython torch==1.2 torchvision==0.4.0 # TODO: fix CI for pytorch>1.2
- pip install -r requirements.txt
- pip install -r tests/requirements.txt
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment